jax.nn.swish

jax.nn.swish(x)

SiLU (又名 swish) 激活函数。

计算逐元素函数

\[\mathrm{silu}(x) = x \cdot \mathrm{sigmoid}(x) = \frac{x}{1 + e^{-x}}\]

swish() 和 silu() 都是同一个函数的别名。

参数:

x (ArrayLike) – 输入数组

返回:

一个数组。

返回类型:

数组

另请参阅

sigmoid()