megengine.functional.equal

equal(x, y)

逐元素相等比较。

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

参数

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

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

返回

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

参见

not_equal, greater_equal, less_equal, greater, less

实际案例

逐元素相等比较：

>>> 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)

上一页

megengine.functional.less_equal

下一页

megengine.functional.not_equal