jax.numpy.isinf¶

jax.numpy.isinf(x)[source]¶

Test element-wise for positive or negative infinity.

LAX-backend implementation of 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