Fonction torch.nansum de PyTorch

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.nansumIl s'agit d'une fonction dans PyTorch permettant de renvoyer la somme des valeurs non-NaN d'un tenseur. Contrairement à sum, elle ignore toutes les valeurs NaN.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

# Renvoyer la somme des valeurs non-NaN
print("Somme des valeurs non-NaN :", torch.nansum(x))

# Somme des valeurs non-NaN le long de dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("dim=0 somme des valeurs non-NaN :", torch.nansum(y, dim=0))
print("dim=1 somme des valeurs non-NaN :", torch.nansum(y, dim=1))

Le résultat de sortie est:

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

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions