Функция 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))

# Сумма не-NaN значений вдоль dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("сумма не-NaN значений по dim=0:", torch.nansum(y, dim=0))
print("сумма не-NaN значений по dim=1:", 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

Другие расширения