函数作用
比较a和b两个数据类型的相同的地方,相同为1,不同为0,注意两个数据类型a,b必须要size相同。
代码
>>> c=torch.tensor([[1,2],[3,4],[5,6]])
>>> d=torch.tensor([[1,1],[2,4],[0,0]])
>>> print('c:', c,'d:', d)
c: tensor([[1, 2],
[3, 4],
[5, 6]])
d: tensor([[1, 1],
[2, 4],
[0, 0]])
>>> a=c.eq(d)
>>> print(a)
tensor([[1, 0],
[0, 1],
[0, 0]], dtype=torch.uint8)
获得相同的个数
>>> a.view(-1).float().sum(0)
tensor(2.)
即c,d中相同的元素个数为2。