函式介紹

a.topk()求a中的最大值或最小值，返回兩個值，一個是a中的值（最大或最小），一個是這個值的索引。

程式碼示例

>>> 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]])
>>> _,largest=False)
>>> pre
tensor([[1],[1],[0]])
>>> _
tensor([[-0.6308],[-3.5483],[-1.4126]])

dim=1，為按行求最大最小值，largest為Ture，求最大值，largest=False，求最小值。

補充知識：python topN 取 最大的N個數 或 最小的N個數

來看程式碼吧！

import numpy as np
a = np.array([1,4,3,5,2])
b = np.argsort(a)
print(b)

print結果[0 4 2 1 3]

說明a[0]最小，a[3]最大

a[0]<a[4]<a[2]<a[1]<a[3]

以上這篇python topk()函式求最大和最小值例項就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。