PyTorch torch.nanmean 関数
Pytorch torch リファレンスマニュアルtorch.nanmeanPyTorch で NaN 値を無視して平均値を計算する関数です。
関数定義
torch.nanmean(input, dim, keepdim=False, out=None)
使用例
実例
import torch
# NaN を含むテンソルを作成
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])
# NaN 以外の値の平均を計算
mean = torch.nanmean(x)
print(非 NaN 平均値:, mean)
# NaN を含むテンソルを作成
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])
# NaN 以外の値の平均を計算
mean = torch.nanmean(x)
print(非 NaN 平均値:, mean)
出力結果は：
非 NaN 均值: tensor(3.)
その他の拡張