pytorch.topk()は、Tensorの最初のk要素とそれらに対応するインデックス値を返すために使用されます。例:
import torch
item=torch.IntTensor([1,2,4,7,3,2])
value,indices=torch.topk(item,3)
print("value:",value)
print("indices:",indices)
出力は次のとおりです。
それらの中で:値は対応するtop3要素に格納され、大から小までの値に従って格納されます
indexは、top3要素のインデックス値を元のTensorの値に格納します