pytorch tensor 元素是否相等

import torch
import numpy as np
x=torch.rand(5,2)
print(x)
print(x[0],x[3])
a=np.array([[2,3],[3,4],[4,5],[6,0]])
a=torch.from_numpy(a)  ####将numpy 转化为tensor
print(a,a.shape)
b=np.array([[2,3],[3,8],[4,5],[3,0]])
b=torch.from_numpy(b)
print(b,b.shape)
print(torch.equal(a,b))
for i in range(0,a.shape[0]):
    print(a[i],b[i])
    if (torch.equal(a[i],b[i])):
         print("66666")