PyTorch torch.nanmedian 関数
Pytorch torch リファレンスマニュアルtorch.nanmedianPyTorch において、テンソルの非NaN値の中央値を返すために使用される関数です。median とは異なり、すべての NaN 値を無視します。
関数定義
torch.nanmedian(input, dim, keepdim=False)
使用例
例
import torch
x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])
# 非NaN値の中央値を返す
print("非NaN中央値:", torch.nanmedian(x))
# dim=0 に沿った非NaN中央値
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("dim=0 非NaN中央値:", torch.nanmedian(y, dim=0))
x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])
# 非NaN値の中央値を返す
print("非NaN中央値:", torch.nanmedian(x))
# dim=0 に沿った非NaN中央値
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("dim=0 非NaN中央値:", torch.nanmedian(y, dim=0))
出力結果は次のとおりです：
非NaN中位数: tensor(2.) dim=0 非NaN中位数: torch.return_types.nanmedian(values=tensor([1., 1., 2.]), indices=tensor([0, 1, 0]))
その他の拡張