您好,登錄后才能下訂單哦!
這篇文章主要講解了“python用topk()函數(shù)怎么求最大最小值”,文中的講解內(nèi)容簡(jiǎn)單清晰,易于學(xué)習(xí)與理解,下面請(qǐng)大家跟著小編的思路慢慢深入,一起來(lái)研究和學(xué)習(xí)“python用topk()函數(shù)怎么求最大最小值”吧!
函數(shù)介紹
a.topk()求a中的最大值或最小值,返回兩個(gè)值,一個(gè)是a中的值(最大或最?。?,一個(gè)是這個(gè)值的索引。
代碼示例
>>> import torch >>> 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為T(mén)ure,求最大值,largest=False,求最小值。
補(bǔ)充知識(shí):python topN 取 最大的N個(gè)數(shù) 或 最小的N個(gè)數(shù)
來(lái)看代碼吧!
import numpy as np a = np.array([1,4,3,5,2]) b = np.argsort(a) print(b)
print結(jié)果[0 4 2 1 3]
說(shuō)明a[0]最小,a[3]最大
a[0]<a[4]<a[2]<a[1]<a[3]
感謝各位的閱讀,以上就是“python用topk()函數(shù)怎么求最大最小值”的內(nèi)容了,經(jīng)過(guò)本文的學(xué)習(xí)后,相信大家對(duì)python用topk()函數(shù)怎么求最大最小值這一問(wèn)題有了更深刻的體會(huì),具體使用情況還需要大家實(shí)踐驗(yàn)證。這里是億速云,小編將為大家推送更多相關(guān)知識(shí)點(diǎn)的文章,歡迎關(guān)注!
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。