【PyTorch学习笔记】torch.ge, torch.gt, torch.le, torch.lt, torch.ne, torch.eq

torch.ge： $\ge$，greater equal，大于等于运
torch.gt： $>$，greater than，大于
torch.le：$\le$，less equal，小于等于
torch.lt：$<$，less than，小于
torch.ne：$\ne$，not equal，不等于
torch.eq：$=$，equal，等于

例1，张量与数值比较：

import torch

input = torch.Tensor([[-1, 0], [1, 2]])

output = torch.ge(input, 1)
print(output)

输出：

tensor([[False, False],
        [ True,  True]])

即 input 中每个元素和 1 进行比较，$\ge$ 1 时返回 True，否则返回 False。

例2，张量与张量比较：

import torch

a = torch.Tensor([[-1, 0], [1, 2]])
b = torch.Tensor([[-1, 0], [2, 3]])

output = torch.ge(a, b)
print(output)

输出：

tensor([[ True,  True],
        [False, False]])

即依次比较 a 和 b 相同位置上的元素，是 $\ge$ 关系时返回 True，否则返回 False。