jax.numpy.equal

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

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

numpy.equal 的 JAX 实现。此函数提供 JAX 数组的 == 运算符的实现。

参数

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

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

返回

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

返回类型:

数组

另请参阅

• jax.numpy.not_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.equal(0., -0.)
Array(True, dtype=bool, weak_type=True)
>>> jnp.equal(1, 1.)
Array(True, dtype=bool, weak_type=True)
>>> jnp.equal(5, jnp.array(5))
Array(True, dtype=bool, weak_type=True)
>>> jnp.equal(2, -2)
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.equal(x, y)
Array([[ True, False, False],
       [False,  True, False],
       [False, False,  True]], dtype=bool)
>>> x == y
Array([[ True, False, False],
       [False,  True, False],
       [False, False,  True]], dtype=bool)