jax.numpy.isposinf

jax.numpy.isposinf(x, /, out=None)

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

JAX 对 numpy.isposinf 的实现。

参数:

x – 输入数组或标量。不支持 complex 数据类型。

返回:

一个与 x 具有相同形状的布尔数组，当 x 为 inf 时包含 True，否则包含 False。

另请参阅

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

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

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

• jax.numpy.isnan(): 返回布尔数组，指示输入的每个元素是否为非数字 (NaN)。

示例

>>> jnp.isposinf(5)
Array(False, dtype=bool)
>>> x = jnp.array([-jnp.inf, 5, jnp.inf, jnp.nan, 1])
>>> jnp.isposinf(x)
Array([False, False,  True, False, False], dtype=bool)