jax.numpy.isinf

jax.numpy.isinf(x, /)

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

numpy.isinf 的 JAX 实现。

参数:

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)