jax.nn.mish

内容

jax.nn.mish#

jax.nn.mish(x)[source]#

Mish 激活函数。

计算逐元素函数

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

更多信息,请参阅 Mish:一种自正则化非单调激活函数

参数:

x (ArrayLike) – 输入数组

返回值:

一个数组。

返回类型:

Array