jax.numpy.isfinite

jax.numpy.isfinite(x, /)

傳回布林陣列，指示輸入的每個元素是否為有限值。

numpy.isfinite 的 JAX 實作。

參數：

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

傳回：

與 x 形狀相同的布林陣列，當 x 不是 inf、-inf 或 NaN 時包含 True，否則包含 False。

傳回型別：

陣列

另請參閱

• jax.numpy.isinf()：傳回布林陣列，指示輸入的每個元素是否為正或負無限大。

• jax.numpy.isposinf()：傳回布林陣列，指示輸入的每個元素是否為正無限大。

• jax.numpy.isneginf()：傳回布林陣列，指示輸入的每個元素是否為負無限大。

• jax.numpy.isnan()：傳回布林陣列，指示輸入的每個元素是否為非數字 (NaN)。

範例

>>> x = jnp.array([-1, 3, jnp.inf, jnp.nan])
>>> jnp.isfinite(x)
Array([ True,  True, False, False], dtype=bool)
>>> jnp.isfinite(3-4j)
Array(True, dtype=bool, weak_type=True)