PyTorch torch.median 関数
Pytorch torch リファレンスマニュアルtorch.medianPyTorch でテンソルの中央値を返すために使用される関数です。中央値は、データを並べ替えた際に中央に位置する値です。
関数定義
torch.median(input, dim, keepdim=False)
使用例
実例
import torch
x = torch.tensor([4, 2, 1, 3, 5])
# すべての要素の中央値を返す
print("全体の中央値:", torch.median(x))
# dim=0 に沿った中央値
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("dim=0 の中央値:", torch.median(y, dim=0))
x = torch.tensor([4, 2, 1, 3, 5])
# すべての要素の中央値を返す
print("全体の中央値:", torch.median(x))
# dim=0 に沿った中央値
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("dim=0 の中央値:", torch.median(y, dim=0))
出力結果は次のとおりです：
全局中位数: tensor(3) dim=0 中位数: torch.return_types.median(values=tensor([1, 1, 2]), indices=tensor([0, 1, 0]))
その他の拡張