jax.numpy.isneginf

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

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

numpy.isneginf 的 JAX 实现。

参数:

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

返回:

一个与 x 形状相同的布尔数组，如果 x 是 -inf，则包含 True，否则包含 False。

另请参阅

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

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

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

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

示例

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