torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le

原创已于 2023-11-02 20:06:41 修改 · 2.9k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#计算机视觉 #深度学习 #机器学习

于 2022-03-09 10:13:42 首次发布

Pytorch 专栏收录该内容

193 篇文章

订阅专栏

这篇博客介绍了PyTorch中用于元素比较的几个函数，如torch.eq、torch.ne、torch.gt、torch.lt、torch.ge和torch.le，通过示例展示了它们如何在张量之间进行逐元素比较并返回布尔张量。这些函数对于张量运算和条件判断非常有用。

部署运行你感兴趣的模型镜像

参考torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le - 云+社区 - 腾讯云

PyTorch - torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le

flyfish

torch.eq、torch.ne、torch.gt、torch.lt、torch.ge、torch.le
以上全是简写
参数是input, other, out=None
逐元素比较input和other
返回是torch.BoolTensor
在这里插入图片描述

import torch

a=torch.tensor([[1, 2], [3, 4]])
b=torch.tensor([[1, 2], [4, 3]])

print(torch.eq(a,b))#equals
# tensor([[ True,  True],
#         [False, False]])

print(torch.ne(a,b))#not equal to
# tensor([[False, False],
#         [ True,  True]])

print(torch.gt(a,b))#greater than
# tensor([[False, False],
#         [False,  True]])

print(torch.lt(a,b))#less than
# tensor([[False, False],
#         [ True, False]])

print(torch.ge(a,b))#greater than or equal to
# tensor([[ True,  True],
#         [False,  True]])

print(torch.le(a,b))#less than or equal to
# tensor([[ True,  True],
#         [ True, False]])

您可能感兴趣的与本文相关的镜像