介绍
这篇文章主要讲解了“python用topk()函数怎么求最大最小值”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“python用topk()函数怎么求最大最小值”吧!
<强>函数介绍强>
a.topk()求一个中的最大值或最小值,返回两个值,一个是一个中的值(最大或最小),一个是这个值的索引。
<强>代码示例强>
在祝辞祝辞import 火炬 在祝辞祝辞,a=torch.randn ((3、5)) ,在的在的在一个 张量([[-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暗=1=True) ,在的在的在前 张量([[2], [4],才能 [3]])才能 在祝辞祝辞_ 张量([[,0.2370], ,0.4145,才能 [-0.0878]])才能 在祝辞祝辞,_,pre=a.topk(最大1暗=1=False) ,在的在的在前 张量([[1], [1],才能 [0]])才能 在祝辞祝辞_ 张量([[-0.6308], [-3.5483],才能 [-1.4126]])才能
暗=1,为按行求最大最小值,最大的为真正的,求最大值,最大=False,求最小值。
<强>补充知识:强> <强> python topN取最大的N个数或最小的N个数强>
来看代码吧!
import numpy as np 时间=a np.array([2] 1、4、3、5日) 时间=b np.argsort (a) 印刷(b)
打印结果(0 4 2 1 3)
说明一个[0]最小,一个[3]最大
[0] & lt; [4] & lt; [2] & lt; [1] & lt;一个[3]
引用>感谢各位的阅读,以上就是“python用topk()函数怎么求最大最小值”的内容了,经过本文的学习后,相信大家对python用topk()函数怎么求最大最小值这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是,小编将为大家推送更多相关知识点的文章,欢迎关注!
python用topk()函数怎么求最大最小值