jax.numpy.isfinite

jax.numpy.isfinite(x, /)

Test element-wise for finiteness (not infinity and not Not a Number).

LAX-backend implementation of numpy.isfinite().

Original docstring below.

The result is returned as a boolean array.

Parameters:

x (array_like) – Input values.

Returns:

y – True where x is not positive infinity, negative infinity, or NaN; false otherwise. This is a scalar if x is a scalar.

Return type:

ndarray, bool