jax.nn.hard_swish

jax.nn.hard_swish(x)

Hard SiLU activation function

Computes the element-wise function

\[\mathrm{hard\_silu}(x) = x \cdot \mathrm{hard\_sigmoid}(x)\]

Parameters:

x (Any) – input array

Return type:

Any