jax.nn.hard_sigmoid

jax.nn.hard_sigmoid(x)[source]

Hard Sigmoid activation function.

Computes the element-wise function

\[\mathrm{hard\_sigmoid}(x) = \frac{\mathrm{relu6}(x + 3)}{6}\]