jax.lax.clamp#
- jax.lax.clamp(min, x, max)[source]#
Elementwise clamp.
Returns \(\mathrm{clamp}(x) = \begin{cases} \mathit{min} & \text{if } x < \mathit{min},\\ \mathit{max} & \text{if } x > \mathit{max},\\ x & \text{otherwise} \end{cases}\).
- Parameters:
min (jax.typing.ArrayLike)
x (jax.typing.ArrayLike)
max (jax.typing.ArrayLike)
- Return type: