jax.numpy.isnan#
- jax.numpy.isnan(x, /)[source]#
Test element-wise for NaN and return result as a boolean array.
LAX-backend implementation of
numpy.isnan()
.Original docstring below.
- Parameters
x (array_like) – Input array.
- Returns
y – True where
x
is NaN, false otherwise. This is a scalar if x is a scalar.- Return type
ndarray or bool