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) – 输入数组

返回:

一个数组。

返回类型:

Array

参见

sigmoid()