megengine.functional.equal

equal(x, y)[源代码]

逐元素判断是否相等。

参数
  • x – 输入张量1。

  • y – 输入张量2。

返回

计算得到的张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor(np.arange(0, 6, dtype=np.float32).reshape(2, 3))
y = tensor(np.arange(0, 6, dtype=np.float32).reshape(2, 3))
out = F.equal(x, y)
print(out.numpy())

输出:

[[1. 1. 1.]
 [1. 1. 1.]]