jax.lax.is_finite

Contents

jax.lax.is_finite#

jax.lax.is_finite(x)[source]#

Elementwise \(\mathrm{isfinite}\).

This function lowers directly to the stablehlo.is_finite operation.

Parameters:: x (ArrayLike) – input array. Must have floating-point type.
Returns:: Array of boolean dtype with the same shape as x, containing False where x is \(\pm\infty\) or \(\mathit{NaN}\), and True otherwise.
Return type:: Array

See also

jax.numpy.isinf(): return True where array is infinite.
jax.numpy.isnan(): return True where array is NaN.