PyTorch torch.mode 関数
Pytorch torch リファレンスマニュアルtorch.modeは、PyTorch でテンソルの最頻値を返すために使用される関数です。最頻値とは、データセット内で最も頻繁に出現する値です。
関数定義
torch.mode(input, dim, keepdim=False)
使用例
例
import torch
x = torch.tensor([1, 2, 2, 3, 3, 3])
# 最頻値を返す
print("最頻値:", torch.mode(x))
# dim=0 に沿った最頻値
y = torch.tensor([[1, 2, 2], [1, 2, 3]])
print("dim=0 の最頻値:", torch.mode(y, dim=0))
# dim=1 に沿った最頻値
print("dim=1 の最頻値:", torch.mode(y, dim=1))
x = torch.tensor([1, 2, 2, 3, 3, 3])
# 最頻値を返す
print("最頻値:", torch.mode(x))
# dim=0 に沿った最頻値
y = torch.tensor([[1, 2, 2], [1, 2, 3]])
print("dim=0 の最頻値:", torch.mode(y, dim=0))
# dim=1 に沿った最頻値
print("dim=1 の最頻値:", torch.mode(y, dim=1))
出力結果は次のとおりです：
众数: torch.return_types.mode(values=tensor(3), indices=tensor(4)) dim=0 众数: torch.return_types.mode(values=tensor([1, 2, 2]), indices=tensor([0, 0, 0])) dim=1 众数: torch.return_types.mode(values=tensor([2, 3]), indices=tensor([1, 2]))
その他の拡張