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)\]

Parameters

x (Any) – input array

Return type

Any