jax.nn.leaky_relu#
- jax.nn.leaky_relu(x, negative_slope=0.01)[source]#
Leaky rectified linear unit activation function.
Computes the element-wise function:
\[\begin{split}\mathrm{leaky\_relu}(x) = \begin{cases} x, & x \ge 0\\ \alpha x, & x < 0 \end{cases}\end{split}\]where \(\alpha\) =
negative_slope
.- Parameters:
- Return type:
- Returns:
An array.
See also