PyTorch torch.kthvalue 関数
Pytorch torch リファレンスマニュアルtorch.kthvaluePyTorch において k 番目に小さい値を返すための関数です。指定した次元における入力テンソルの k 番目の最小値とそのインデックスを返します。
関数定義
torch.kthvalue(input, k, dim=None, keepdim=False)
使用例
実例
import torch
# テンソルを作成
x = torch.tensor([3, 1, 2, 4])
# 2番目に小さい値とインデックスを返す
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
# テンソルを作成
x = torch.tensor([3, 1, 2, 4])
# 2番目に小さい値とインデックスを返す
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
出力結果は次のとおりです：
tensor(2) tensor(2)
その他の拡張