Ptorch中关系运算符lt、gt、eq、le、ge的用法

Ptorch中关系运算符lt、gt、eq、le、ge的用法

ptorch中的关系运算符lt（小于）、gt（大于）、eq（等于）、le（小于等于）、ge（大于等于），这里拿lt（小于）举例

import torch

import torch

a =torch.Tensor([1,2,4,4,5])

print(a<4)
print(torch.lt(a,4)) #lt gt eq le ge
print(a[a<4])

a是一个Tensor类型的数据，lt(a,4)：取出a<4的掩码（因为ptorch里面没有True或False，所以称为掩码）
print(torch.lt(a,4))：
而print(a<4)的结果和上面的一样。