jax.numpy.isnan

jax.numpy.isnan(x, /)

返回一個布林陣列，指示輸入的每個元素是否為 NaN。

JAX 實作的 numpy.isnan。

參數：

x (ArrayLike) – 輸入陣列或純量。

返回：

與 x 形狀相同的布林陣列，若 x 不是數字 (即 NaN) 則包含 True，否則包含 False。

返回類型：

Array

參見

• 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)