Return a boolean array indicating whether each element of input is infinite.
JAX implementation of numpy.isinf
.
x (ArrayLike) – input array or scalar.
A boolean array of same shape as x
containing True
where x
is inf
or -inf
, and False
otherwise.
See also
jax.numpy.isposinf()
: Returns a boolean array indicating whether each element of input is positive infinity.
jax.numpy.isneginf()
: Returns a boolean array indicating whether each element of input is negative infinity.
jax.numpy.isfinite()
: Returns a boolean array indicating whether each element of input is finite.
jax.numpy.isnan()
: Returns a boolean array indicating whether each element of input is not a number (NaN
).
Examples
>>> jnp.isinf(jnp.inf) Array(True, dtype=bool) >>> x = jnp.array([2+3j, -jnp.inf, 6, jnp.inf, jnp.nan]) >>> jnp.isinf(x) Array([False, True, False, True, False], dtype=bool)
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.3