jax.lax.is_finite

jax.lax.is_finite(x)[source]

Elementwise \(\mathrm{isfinite}\).

For each element x returns True if and only if x is not \(\pm\infty\) or \(\mathit{NaN}\).

Parameters

x (Any) –

Return type

Any