1. torch.gt(), torch.le(), torch.eq()用法介绍
?pytorch中逐元素比较两个torch.tensor大小用到了如下的三个函数：
- torch.gt(tensor1, tensor2, out=None) || tensor1.gt(tensor2,out=None)：tensor1对应的元素大于tensor2的元素会返回True，否则返回False。参数out表示为一个数或者是与第一个参数相同形状和类型的tensor。
- torch.lt(tensor1, tensor2, out=None) || tensor1.lt(tensor2,out=None)：tensor1对应的元素小于tensor2的元素会返回True，否则返回False。参数out表示为一个数或者是与第一个参数相同形状和类型的tensor。
- torch.eq(tensor1, tensor2, out=None) || tensor1.eq(tensor2,out=None)：tensor1对应的元素等于tensor2的元素会返回True，否则返回False。参数out表示为一个数或者是与第一个参数相同形状和类型的tensor。
返回值为一个torch.ByteTensor张量，包含了每个位置的比较结果。
2.程序代码
>>> import torch
>>> Matrix_A = torch.tensor([1,2,3,4,5,3,1])
>>> Matrix_B = torch.tensor([2,1,3,1,5,2,0])
>>> print(torch.gt(Matrix_A,Matrix_B))
tensor([False, True, False, True, False, True, True])
>>> print(torch.lt(Matrix_A,Matrix_B))
tensor([ True, False, False, False, False, False, False])
>>> print(torch.eq(Matrix_A,Matrix_B))
tensor([False, False, True, False, True, False, False])
>>> print(Matrix_A.gt(Matrix_B))
tensor([False, True, False, True, False, True, True])
>>> print(Matrix_A.lt(Matrix_B))
tensor([ True, False, False, False, False, False, False])
>>> print(Matrix_A.eq(Matrix_B))
tensor([False, False, True, False, True, False, False])
|