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