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]])