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