torch.eq

torch.eq(input, other, *, out=None) → Tensor

计算逐元素相等性

第二个参数可以是与第一个参数可广播的数字或张量。

参数:

• 输入（张量）- 要比较的张量

• 其他（张量或浮点数）- 要比较的张量或值

关键字参数:

输出（张量，可选）- 输出张量。

返回值:

一个布尔张量，当 input 等于 other 时为 True，其他地方为 False

示例：

>>> torch.eq(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]]))
tensor([[ True, False],
        [False, True]])