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返回的結果一樣