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()