jax.lax.clamp

jax.lax.clamp(min, x, max)

逐元素钳制。

返回 \(\mathrm{clamp}(x) = \begin{cases} \mathit{min} & \text{如果 } x < \mathit{min},\\ \mathit{max} & \text{如果 } x > \mathit{max},\\ x & \text{否则} \end{cases}\).

参数:

• min (ArrayLike)

• x (ArrayLike)

• max (ArrayLike)

返回类型:

Array