torch.ge,torch.gt,torch.le

torch.ge,torch.gt,torch.le

gt: 大于

ge：大于等于

lt: 小于

le：小于等于

if __name__ == '__main__':

    import torch
    data=torch.Tensor([[0,1, 2], [-1,3, 4]])

    bb=data.ge(0)
    print(bb)
           

判断相等：

if __name__ == '__main__':

    import torch
    data=torch.Tensor([[0,1, 2], [-1,3, 4]]).cuda()

    bb=data.__eq__(0)
    print(bb)
           

torch.ge
torch.ge(input, other, out=None) → Tensor
逐元素比较input和other，即是否 input>=otherinput>=other。

如果两个张量有相同的形状和元素值，则返回True ，否则 False。 第二个参数可以为一个数或与第一个参数相同形状和类型的张量

参数:

input (Tensor) – 待对比的张量
other (Tensor or float) – 对比的张量或float值
out (Tensor, optional) – 输出张量。必须为ByteTensor或者与第一个参数tensor相同类型。
返回值： 一个 torch.ByteTensor 张量，包含了每个位置的比较结果(是否 input >= other