jax.nn.relu

jax.nn.relu = <jax._src.custom_derivatives.custom_jvp 对象>

整流线性单元激活函数。

计算逐元素函数

\[\mathrm{relu}(x) = \max(x, 0)\]

除了在微分时，我们取

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

有关更多信息，请参阅 ReLU’(0) 对反向传播的数值影响。

参数:

x (类数组) – 输入数组

返回:

一个数组。

返回类型:

数组

示例

>>> jax.nn.relu(jax.numpy.array([-2., -1., -0.5, 0, 0.5, 1., 2.]))
Array([0. , 0. , 0. , 0. , 0.5, 1. , 2. ], dtype=float32)

另请参阅

relu6()