pytorch学习笔记2：比较操作

 

目录

 

torch.eq

torch.equal

torch.ge

torch.gt

---

torch.eq

torch.eq(input, other, out=None) → Tensor

比较元素相等性。第二个参数可为一个数或与第一个参数同类型形状的张量。

参数：

• input (Tensor) – 待比较张量
• other (Tensor or float) – 比较张量或数
• out (Tensor, optional) – 输出张量，须为 ByteTensor类型 or 与input同类型

返回值： 一个 torch.ByteTensor 张量，包含了每个位置的比较结果(相等为1，不等为0 )

返回类型： Tensor

例子：

>>> torch.eq(torch.Tensor([[1, 2], [3, 4]]), torch.Tensor([[1, 1], [4, 4]]))
1  0
0  1
[torch.ByteTensor of size 2x2]

torch.equal

torch.equal(tensor1, tensor2) → bool

如果两个张量有相同的形状和元素值，则返回True ，否则 False。

例子：

>>> torch.equal(torch.Tensor([1, 2]), torch.Tensor([1, 2]))
True
>>>a=torch.Tensor([[1,2,3],[4,5,6]])
b=torch.Tensor([[1,2,3],[4,5,6]])
c=torch.Tensor([[1,2,3],[1,2,3]])
print(torch.equal(a,b))
print(torch.equal(a,c))

True
False

torch.ge

torch.ge(input, other, out=None) → Tensor

逐元素比较input和other，即是否 input>=otherinput>=other。也就是第二个张量的元素是否小于等于第一个相应位置的元素，满足条件为1，否则为0

例子：

>>> torch.ge(torch.Tensor([[1, 2], [3, 4]]), torch.Tensor([[1, 1], [4, 4]]))
 1  1
 0  1
[torch.ByteTensor of size 2x2]

torch.gt

torch.gt(input, other, out=None) → Tensor

逐元素比较input和other ， 即是否input>otherinput>other 。 第二个参数可以为一个数或与第一个参数相同形状和类型的张量

第二个张量的元素是否小于第一个相应位置的元素，满足条件为1，否则为0

例子：

>>> torch.gt(torch.Tensor([[1, 2], [3, 4]]), torch.Tensor([[1, 1], [4, 4]]))
 0  1
 0  0
[torch.ByteTensor of size 2x2]