jax.nn.mish

Contents

jax.nn.mish#

jax.nn.mish(x)#

Mish activation function.

Computes the element-wise function:

\[\mathrm{mish}(x) = x \cdot \mathrm{tanh}(\mathrm{softplus}(x))\]

For more information, see Mish: A Self Regularized Non-Monotonic Activation Function.

Parameters:

x (jax.typing.ArrayLike) – input array

Returns:

An array.

Return type:

Array