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

发布了18 篇原创文章 · 获赞 2 · 访问量 342

猜你喜欢

转载自blog.csdn.net/weixin_44928646/article/details/104628619