pytorch的比较操作

1.torch.eq(input, other, out =None)

比较元素是否相等，other可以是一个数，或者跟input同类型同维度的张量
参数

• input(Tensor)：待比较的张量
• other：比较的张量或者数
• out：（可选），输出张量

2.torch.equal(tensor1, tensor2, out =None)

判断两个张量是否有相同的形状和元素值

3.torch.gt() 绝对大于

参数同1
比较a中元素大于（这里是严格大于）b中对应元素，大于则为true，不大于则为false，这里a为Tensor，b可以为与a的size相同的Tensor或常数。

4.torch.ge() 大于等于

5.torch.lt() 绝对小于

6.torch.le() 小于等于

例子如下：

import torch
a = torch.Tensor([[1,2,3],
                 [3,2,1]])

b = torch.Tensor([[1,2,3],
                 [3,2,2]])
print(torch.eq(a,b)) #每个元素
print(torch.equal(a,b))#tensor形状及所有元素值

print(torch.gt(a,b))#绝对大于
print(torch.gt(a,2))

print(torch.ge(a,b))#大于等于
print(torch.ge(a,2))

print(torch.lt(a,b))#绝对小于
print(torch.lt(a,2))

print(torch.le(a,b))#小于等于
print(torch.le(a,2))

具体的可参考这篇博客，写的很详细
Pytorch学习之torch----比较操作(Comparison Ops)

附，类似操作

import torch
a = torch.Tensor([1,2,3,4,5])

print(torch.lt(a,4))
print(torch.masked_select(a, a<4)) ##符合条件的元素
print(torch.nonzero(a<4))##符合条件的元素所对应的索引
#输出结果
tensor([ True,  True,  True, False, False])
tensor([1., 2., 3.])
tensor([[0],
        [1],
        [2]])