torch.ge,torch.gt,torch.le
torch.ge
torch.ge(input, other, out=None) → Tensor
逐元素比较input和other,即是否 input>=otherinput>=other。
如果两个张量有相同的形状和元素值,则返回True ,否则 False。 第二个参数可以为一个数或与第一个参数相同形状和类型的张量
参数:
input (Tensor) – 待对比的...
原创
2019-01-11 16:41:36 ·
1122 阅读 ·
1 评论