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
out (ndarray, None, or tuple of ndarray and None, optional) – A location into which the result is stored. If provided, it must have a shape that the inputs broadcast to. If not provided or None, a freshly-allocated array is returned. A tuple (possible only as a keyword argument) must have length equal to the number of outputs.
where (array_like, optional) – This condition is broadcast over the input. At locations where the condition is True, the out array will be set to the ufunc result. Elsewhere, the out array will retain its original value. Note that if an uninitialized out array is created via the default
out=None, locations within it where the condition is False will remain uninitialized.**kwargs – For other keyword-only arguments, see the ufunc docs.
- Returns
y – True where
xis positive or negative infinity, false otherwise. This is a scalar if x is a scalar.- Return type
bool (scalar) or boolean ndarray