jax.numpy.isinf

jax.numpy.isinf(x, /)

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

JAX 實作的 numpy.isinf。

參數：:

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

返回：:

一個與 x 形狀相同的布林陣列，當 x 為 inf 或 -inf 時包含 True，否則包含 False。

返回類型：:

陣列

另請參閱

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

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

• jax.numpy.isfinite()：返回一個布林陣列，指示輸入的每個元素是否為有限數。

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

範例

>>> jnp.isinf(jnp.inf)
Array(True, dtype=bool)
>>> x = jnp.array([2+3j, -jnp.inf, 6, jnp.inf, jnp.nan])
>>> jnp.isinf(x)
Array([False,  True, False,  True, False], dtype=bool)