PyTorch torch.topk 関数
Pytorch torch リファレンスマニュアルtorch.topkPyTorch において、最大の k 個の要素とインデックスを返すために使用される関数です。
関数定義
torch.topk(input, k, dim, largest, sorted, out)
使用例
例
import torch
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# 最大の 2 つの要素を返す
values, indices = torch.topk(x, k=2)
print("最大の 2 つの値:")
print(values)
print("対応するインデックス:")
print(indices)
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# 最大の 2 つの要素を返す
values, indices = torch.topk(x, k=2)
print("最大の 2 つの値:")
print(values)
print("対応するインデックス:")
print(indices)
出力結果は次のとおりです：
最大的 2 个值:
tensor([[3, 2],
[6, 5]])
对应的索引:
tensor([[0, 2],
[0, 2]])
その他の拡張