jax.numpy.equal

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

逐元素返回 (x1 == x2)。

LAX 后端实现 numpy.equal()。

原始文档字符串如下。

返回：

out – 输出数组，x1 和 x2 的逐元素比较。 通常为布尔类型，除非传递 dtype=object。 如果 x1 和 x2 都是标量，则为标量。

返回类型：

ndarray 或标量

参数：

• x (ArrayLike)

• y (ArrayLike)