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()