jax.nn.selu

目录

jax.nn.selu#

jax.nn.selu(x)[源代码][源代码]#

缩放指数线性单元激活。

计算逐元素函数:

\[\begin{split}\mathrm{selu}(x) = \lambda \begin{cases} x, & x > 0\\ \alpha e^x - \alpha, & x \le 0 \end{cases}\end{split}\]

其中 \(\lambda = 1.0507009873554804934193349852946\)\(\alpha = 1.6732632423543772848170429916717\)

更多信息,请参见 自归一化神经网络

参数:

x (ArrayLike) – 输入数组

返回:

一个数组。

返回类型:

Array

参见

elu()