jax.numpy.isinf¶
-
jax.numpy.
isinf
(x)[source]¶ Test element-wise for positive or negative infinity.
LAX-backend implementation of
isinf()
. Original docstring below.isinf(x, /, out=None, *, where=True, casting=’same_kind’, order=’K’, dtype=None, subok=True[, signature, extobj])
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
See also
Notes
NumPy uses the IEEE Standard for Binary Floating-Point for Arithmetic (IEEE 754).
Errors result if the second argument is supplied when the first argument is a scalar, or if the first and second arguments have different shapes.
Examples
>>> np.isinf(np.inf) True >>> np.isinf(np.nan) False >>> np.isinf(np.NINF) True >>> np.isinf([np.inf, -np.inf, 1.0, np.nan]) array([ True, True, False, False])
>>> x = np.array([-np.inf, 0., np.inf]) >>> y = np.array([2, 2, 2]) >>> np.isinf(x, y) array([1, 0, 1]) >>> y array([1, 0, 1])