jax.numpy.isinf

jax.numpy.isinf(x, /)

Test element-wise for positive or negative infinity.

LAX-backend implementation of numpy.isinf().

Original docstring below.

Returns a boolean array of the same shape as x, True where x == +/-inf, otherwise False.

Parameters:

x (array_like) – Input values

Returns:

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

Return type:

bool (scalar) or boolean ndarray