python topk()函数求最大和最小值实例
函数介绍
a.topk()求a中的最大值或最小值,返回两个值,一个是a中的值(最大或最小),一个是这个值的索引。
代码示例
>>>importtorch >>>a=torch.randn((3,5)) >>>a tensor([[-0.4790,-0.6308,0.2370,0.0380,-0.0579], [-0.6712,-3.5483,-0.2370,-0.8658,0.4145], [-1.4126,-0.8786,-0.4216,-0.0878,-1.4015]]) >>>_,pre=a.topk(1,dim=1,largest=True) >>>pre tensor([[2], [4], [3]]) >>>_ tensor([[0.2370], [0.4145], [-0.0878]]) >>>_,pre=a.topk(1,dim=1,largest=False) >>>pre tensor([[1], [1], [0]]) >>>_ tensor([[-0.6308], [-3.5483], [-1.4126]])
dim=1,为按行求最大最小值,largest为Ture,求最大值,largest=False,求最小值。
补充知识:pythontopN取最大的N个数或最小的N个数
来看代码吧!
importnumpyasnp a=np.array([1,4,3,5,2]) b=np.argsort(a) print(b)
print结果[04213]
说明a[0]最小,a[3]最大
以上这篇pythontopk()函数求最大和最小值实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。