torch.eq()

若predicted与label.data对应数值相同，则torch.eq()返回1，否则返回0，张量类型

 torch.eq(predicted,label.data).sum() 返回一个张量，张量值为对应值相同的个数

 torch.eq(predicted,label.data).sum().iten()返回值为torch.eq()所有值相加

import torch

predict_labels = torch.tensor([0,1 ,2 ,3, 4,6,7])
labels = torch.tensor([4, 3 ,2, 1 ,4,6,7])


correct = torch.eq(predict_labels,labels)
correct1 = torch.eq(predict_labels,labels).sum()
correct2 = torch.eq(predict_labels,labels).sum().item()


print(correct)
'''
tensor([False, False,  True, False,  True,  True,  True])
'''
print(correct1)  #生成张量
'''
tensor(4)
'''
print(correct2)  #数值相加
'''
4
'''