PyTorch torch.msort 関数
Pytorch torch リファレンスマニュアルtorch.msortPyTorch において、ソートされたテンソルを返すために使用される関数です。入力テンソルを昇順に並べ替えたコピーを返します。テンソルを flattened にしてからソートし、再度 reshape するのに似ています。
関数定義
torch.msort(input)
使用例
例
import torch
# テンソルを作成
x = torch.tensor([3, 1, 2])
# ソートされたテンソルを返す
result = torch.msort(x)
print(result)
# テンソルを作成
x = torch.tensor([3, 1, 2])
# ソートされたテンソルを返す
result = torch.msort(x)
print(result)
出力結果は次のとおりです：
tensor([1, 2, 3])
その他の拡張