python用topk()函数怎么求最大最小值

  介绍

这篇文章主要讲解了“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()函数怎么求最大最小值