jax.nn.relu

jax.nn.relu = <jax._src.custom_derivatives.custom_jvp object>

整流線性單元啟動函數。

計算逐元素函數

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

微分下除外，我們取

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

如需更多資訊，請參閱 ReLU’(0) 對反向傳播的數值影響。

參數:

x (ArrayLike) – 輸入陣列

傳回:

一個陣列。

傳回類型:

Array

範例

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