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)的结果和上面的一样。