pytorch比较Tensor

 

pytorch比较数组或矩阵长度：

if tx[conf_mask >= 1].size()>torch.Size([0]):

 

1，比较大小：

对于t o r c h . a l l c l o s e ( ) torch.allclose()torch.allclose()函数，比较的是两个元素是否接近，比较A和B是否接近的公式为：
∣A−B∣<=atol+rtol×∣B∣
 

import torch

# 比较两个数是否接近
a = torch.tensor([10.0])
b = torch.tensor([10.1])
print(torch.allclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False))
print(torch.allclose(a, b, rtol=0.1, atol=0.01, equal_nan=False))
# 在不同的判断条件下，得出的结果也是不同的

# 如果equal_nan=True，那么缺失值可以判断为接近
a = torch.tensor(float("nan"))
print(torch.allclose(a, a, equal_nan=False))
print(torch.allclose(a, a, equal_nan=True))


a = torch.tensor([1, 2, 3, 4, 5, 6])
b = torch.arange(1, 7)
c = torch.unsqueeze(b, dim=0)
# torch.eq()函数用来判断两个元素是否相等
p