jax.nn.relu

Warning

This page was created from a pull request (#9655).

jax.nn.relu¶

jax.nn.relu(x) = <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

previous

jax.nn.initializers.he_normal

next

jax.nn.relu6