本篇内容主要讲解“PyTorch中的参数怎么使用”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“PyTorch中的参数怎么使用”吧!
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]])
到此，相信大家对“PyTorch中的参数怎么使用”有了更深的了解，不妨来实际操作一番吧！这里是亿速云网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
