jax.nn.selu

内容

jax.nn.selu#

jax.nn.selu(x)[source]#

缩放指数线性单元激活。

计算逐元素函数

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

返回值:

一个数组。

返回值类型:

数组

另请参阅

elu()