jax.numpy.equal

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

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

JAX 實作的 numpy.equal。此函式為 JAX 陣列的 == 運算子提供實作。

參數：

• x (ArrayLike) – 輸入陣列或純量。

• y (ArrayLike) – 輸入陣列或純量。x 和 y 應具有相同形狀或可廣播相容。

傳回：

布林陣列，包含 True，若 x == y 的元素為真，否則為 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)