jax.numpy.isneginf#

jax.numpy.isneginf(x, /, out=None)[源代码]#

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

JAX 实现的 numpy.isneginf

参数:

x – 输入数组或标量。不支持 complex dtype。

返回值:

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)