jax.nn.elu

目录

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