如何解决PyTorch方法返回一维张量中具有最高计数的元素?
是否有内置的PyTorch方法采用一维张量,并在张量中返回计数最高的元素?
例如,如果我们输入torch.tensor([2,2,3,4,5])
,则该方法应返回2
,因为它出现的次数最多。如果出现频率关联,则应返回值较低的元素;输入torch.tensor([1,1,5])
应该返回1
。
请明确一点,我只想知道是否有一个现成的内置PyTorch方法可以做到这一点。如果没有这种方法,请不要发布解决方案,因为我想尝试自行解决。
解决方法
是的torch.mode()
是内置函数(读为here),可以处理您的两种情况。
torch.mode(alpha,0) #alpha being the name of tensor
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。