flax.linen.PReLU¶

class flax.linen.PReLU(param_dtype=<class 'jax.numpy.float32'>, negative_slope_init=0.01, parent=<flax.linen.module._Sentinel object>, name=None)[source]¶

Parametric Rectified Linear Unit (PReLU) activation function.

Parameters

param_dtype (Any) –
negative_slope_init (float) –
parent (Optional[Union[Type[flax.linen.module.Module], Type[flax.core.scope.Scope], Type[flax.linen.module._Sentinel]]]) –
name (str) –

Return type

None

param_dtype¶

the dtype passed to parameter initializers (default: float32).

Type: Any

negative_slope_init¶

the value to initialize the negative slope (default 0.01).

Type: float

__init__(param_dtype=<class 'jax.numpy.float32'>, negative_slope_init=0.01, parent=<flax.linen.module._Sentinel object>, name=None)¶

Parameters

param_dtype (Any) –
negative_slope_init (float) –
parent (Optional[Union[Type[flax.linen.module.Module], Type[flax.core.scope.Scope], Type[flax.linen.module._Sentinel]]]) –
name (Optional[str]) –

Return type

None

Methods

`__init__`([param_dtype, negative_slope_init, ...])
`apply`(variables, *args[, rngs, method, ...])	Applies a module method to variables and returns output and modified variables.
`bind`(variables, *args[, rngs, mutable])	Creates an interactive Module instance by binding variables and RNGs.
`clone`(*[, parent])	Creates a clone of this Module, with optionally updated arguments.
`get_variable`(col, name[, default])	Retrieves the value of a Variable.
`has_rng`(name)	Returns true if a PRNGSequence with name name exists.
`has_variable`(col, name)	Checks if a variable of given collection and name exists in this Module.
`init`(rngs, *args[, method, mutable])	Initializes a module method with variables and returns modified variables.
`init_with_output`(rngs, *args[, method, mutable])	Initializes a module method with variables and returns output and modified variables.
`is_mutable_collection`(col)	Returns true if the collection col is mutable.
`make_rng`(name)	Returns a new RNG key from a given RNG sequence for this Module.
`param`(name, init_fn, *init_args)	Declares and returns a parameter in this Module.
`put_variable`(col, name, value)	Sets the value of a Variable.
`setup`()	Initializes a Module lazily (similar to a lazy `__init__`).
`sow`(col, name, value[, reduce_fn, init_fn])	Stores a value in a collection.
`variable`(col, name[, init_fn])	Declares and returns a variable in this Module.

Attributes

`name`
`negative_slope_init`
`parent`
`scope`
`variables`	Returns the variables in this module.