jax.nn.relu6

Warning

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

jax.nn.relu6¶

jax.nn.relu6(x)[source]¶

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

previous

jax.nn.relu

next

jax.nn.sigmoid