jax.nn.relu6

目录

jax.nn.relu6#

jax.nn.relu6 = <jax._src.custom_derivatives.custom_jvp object>[源代码][源代码]#

校正线性单元6激活函数。

计算逐元素函数

\[\mathrm{relu6}(x) = \min(\max(x, 0), 6)\]

除了在微分下,我们取:

\[\nabla \mathrm{relu}(0) = 0\]

\[\nabla \mathrm{relu}(6) = 0\]
参数:

x (ArrayLike) – 输入数组

返回:

一个数组。

返回类型:

Array

参见

relu()