jax.numpy.isnan

jax.numpy.isnan(x, /)

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