jax.numpy.isnan

jax.numpy.isnan(x, /)

测试 NaN 的逐元素，并将结果作为布尔数组返回。

LAX 后端实现 numpy.isnan()。

原始文档字符串如下。

参数：:

x (array_like) – 输入数组。

返回值：:

y – 如果 x 是 NaN，则为 True，否则为 False。 如果 x 是标量，则为标量。

返回类型：:

ndarray 或 bool