pytorch中比较两个tensor是否相等

1. tensor.equal()方法：

# 该方法用于比较两个tensor是否一样，一样则返回True否则为False
a = torch.tensor([1,2,3,4])
b = torch.tensor([1,2,3,4])
print(a.equal(b))    # 返回True

2. 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返回的结果一样

pytorch中比较两个tensor是否相等

原文：https://www.cnblogs.com/noob-l/p/14648833.html