pytorch中比较两个tensor是否相等
- tensor.equal()方法：
# 该方法用于比较两个tensor是否一样，一样则返回True否则为False
a = torch.tensor([1,2,3,4])
b = torch.tensor([1,2,3,4])
print(a.equal(b)) # 返回True
- tensor.eq()方法：
# 该方法用于主元素比较是否相等，相等则在对应位置返回True，否则为False
a = torch.tensor([1,2,2,3])
b = torch.tensor([2,2,3,3])
print(a.eq(b)) # 返回tensor([False,True,False,True]),与a==b返回的结果一样
