jax.nn.leaky_relu

jax.nn.leaky_relu(x, negative_slope=0.01)

泄漏整流线性单元激活函数。

计算逐元素函数：

\[\begin{split}\mathrm{leaky\_relu}(x) = \begin{cases} x, & x \ge 0\\ \alpha x, & x < 0 \end{cases}\end{split}\]

其中 \(\alpha\) = negative_slope。

参数:

• x (ArrayLike) – 输入数组

• negative_slope (ArrayLike) – 指定负斜率的数组或标量（默认值：0.01）

返回:

一个数组。

返回类型:

Array

参见

relu()