jax.nn.relu6

内容

jax.nn.relu6#

jax.nn.relu6 = <jax._src.custom_derivatives.custom_jvp object>[source]#

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

计算逐元素函数

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

除了在微分下,我们取

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

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

x (ArrayLike) – 输入数组

返回值:

一个数组。

返回类型:

数组

另请参阅

relu()