jax.nn.elu

jax.nn.elu(x, alpha=1.0)

指数线性单元激活函数。

计算逐元素函数

\[\begin{split}\mathrm{elu}(x) = \begin{cases} x, & x > 0\\ \alpha \left(\exp(x) - 1\right), & x \le 0 \end{cases}\end{split}\]

参数：

• x (ArrayLike) – 输入数组

• alpha (ArrayLike) – alpha 值的标量或数组（默认值：1.0）

返回值：

一个数组。

返回值类型：

Array

另请参阅

selu()