PyTorch torch.nansum-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.nansumEs ist eine Funktion in PyTorch, die die Summe der Nicht-NaN-Werte eines Tensors zurückgibt. Im Gegensatz zu sum ignoriert sie alle NaN-Werte.

Funktionsdefinition

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

Anwendungsbeispiel

Beispiel

import torch

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

# Summe der Nicht-NaN-Werte zurückgeben
print("Summe der Nicht-NaN-Werte:", torch.nansum(x))

# Summe der Nicht-NaN-Werte entlang dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("Summe der Nicht-NaN-Werte entlang dim=0:", torch.nansum(y, dim=0))
print("Summe der Nicht-NaN-Werte entlang dim=1:", torch.nansum(y, dim=1))

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen