Warning

This page was created from a pull request.

jax.nn.relu6

jax.nn.relu6(x)

Rectified Linear Unit 6 activation function.

Computes the element-wise function

\[\mathrm{relu6}(x) = \min(\max(x, 0), 6)\]