jax.numpy.isnan

jax.numpy.isnan(x, /)

返回一个布尔数组，指示输入的每个元素是否为 NaN。

numpy.isnan 的 JAX 实现。

参数:

x (类数组) – 输入数组或标量。

返回:

一个与 x 形状相同的布尔数组，其中 x 不是数字（即 NaN）的位置为 True，否则为 False。

返回类型:

数组

另请参阅

• jax.numpy.isfinite()：返回一个布尔数组，指示输入的每个元素是否为有限值。

• jax.numpy.isinf()：返回一个布尔数组，指示输入的每个元素是否为正无穷大或负无穷大。

• jax.numpy.isposinf()：返回一个布尔数组，指示输入的每个元素是否为正无穷大。

• jax.numpy.isneginf()：返回一个布尔数组，指示输入的每个元素是否为负无穷大。

示例

>>> jnp.isnan(6)
Array(False, dtype=bool, weak_type=True)
>>> x = jnp.array([2, 1+4j, jnp.inf, jnp.nan])
>>> jnp.isnan(x)
Array([False, False, False,  True], dtype=bool)