jax.nn.hard_sigmoid

Contents

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}\]
Parameters:

x (jax.typing.ArrayLike) – input array

Returns:

An array.

Return type:

Array

See also

relu6()