PyTorch torch.nansum 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.nansumは、PyTorch でテンソルの非NaN値の合計を返すために使用される関数です。sum とは異なり、すべての NaN 値を無視します。

関数定義

torch.nansum(input, dim, keepdim=False)

使用例

例

import torch

x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])

# 非NaN値の合計を返す
print("非NaN値の合計:", torch.nansum(x))

# dim=0 に沿った非NaN値の合計
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("dim=0 の非NaN値の合計:", torch.nansum(y, dim=0))
print("dim=1 の非NaN値の合計:", torch.nansum(y, dim=1))

出力結果は次のとおりです：

非NaN值之和: tensor(11.)
dim=0 非NaN值之和: tensor([5., 1., 5.])
dim=1 非NaN值之和: tensor([3., 8.])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張