megengine.functional.equal

equal(x, y)[源代码]

逐元素相等比较。

计算 \(x_i == y_i\) 的值,其中 \(x_i\) 为输入张量 \(x\) 的遍历,而 \(y_i\) 为输入张量 \(y\) 相应位置的元素。

参数
  • x – 第一个输入张量。支持任意数据类型。

  • y – 第二个输入张量. Laytout必须跟 \(x\) 兼容(具体参见 广播机制与规则 )。支持任意数据类型。

返回

承接逐元素操作结果的张量。返回的张量必须是 bool 数据类型。

实际案例

逐元素相等比较:

>>> x = Tensor([1, 2, 3])
>>> y = Tensor([1, 2, 4])
>>> F.equal(x, y)
Tensor([ True  True False], dtype=bool, device=xpux:0)

当输入张量是布尔类型,则可以使用 == 运算符作为 F.equal 的等价表达。

>>> x == y
Tensor([ True  True False], dtype=bool, device=xpux:0)