flax.linen.relu

flax.linen.relu = <jax._src.custom_derivatives.custom_jvp object>[source]

Rectified linear unit activation function.

Computes the element-wise function:

\[\mathrm{relu}(x) = \max(x, 0)\]
Parameters

x (Any) – input array

Return type

Any