Toggle navigation

.rst .pdf

Warning

This page was created from a pull request (#9655).

jax.lax.clamp

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

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

• x (Any) –

• max (Any) –

Return type

Any

previous

jax.lax.ceil

next

jax.lax.collapse