jax.numpy.isnan¶

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

Test element-wise for NaN and return result as a boolean array.

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