jax.numpy.not_equal

jax.numpy.not_equal(x, y, /)

返回逐元素的 x != y 真值。

numpy.not_equal 的 JAX 实现。此函数为 JAX 数组提供了 != 运算符的实现。

参数:

• x (ArrayLike) – 输入数组或标量。

• y (ArrayLike) – 输入数组或标量。x 和 y 应具有相同的形状或广播兼容。

返回:

一个布尔数组，其中 x != y 的元素为 True，否则为 False。

返回类型:

数组

另请参阅

• jax.numpy.equal(): 返回 x == y 的逐元素真值。

• jax.numpy.greater_equal(): 返回 x >= y 的逐元素真值。

• jax.numpy.less_equal(): 返回 x <= y 的逐元素真值。

• jax.numpy.greater(): 返回 x > y 的逐元素真值。

• jax.numpy.less(): 返回 x < y 的逐元素真值。

示例

>>> jnp.not_equal(0., -0.)
Array(False, dtype=bool, weak_type=True)
>>> jnp.not_equal(-2, 2)
Array(True, dtype=bool, weak_type=True)
>>> jnp.not_equal(1, 1.)
Array(False, dtype=bool, weak_type=True)
>>> jnp.not_equal(5, jnp.array(5))
Array(False, dtype=bool, weak_type=True)
>>> x = jnp.array([[1, 2, 3],
...                [4, 5, 6],
...                [7, 8, 9]])
>>> y = jnp.array([1, 5, 9])
>>> jnp.not_equal(x, y)
Array([[False,  True,  True],
       [ True, False,  True],
       [ True,  True, False]], dtype=bool)
>>> x != y
Array([[False,  True,  True],
       [ True, False,  True],
       [ True,  True, False]], dtype=bool)