PyTorch torch.nanmean Funktion
PyTorch torch-Referenzhandbuch
torch.nanmeanEs ist eine Funktion in PyTorch, die verwendet wird, um den Durchschnittswert zu berechnen und dabei NaN-Werte zu ignorieren.
Funktionsdefinition
torch.nanmean(input, dim, keepdim=False, out=None)
Anwendungsbeispiel
Beispiel
import torch
# Einen Tensor erstellen, der NaN enthält
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])
# Mittelwert der Nicht-NaN-Werte berechnen
mean = torch.nanmean(x)
print("Nicht-NaN-Mittelwert:", mean)
# Einen Tensor erstellen, der NaN enthält
x = torch.tensor([1.0, 2.0, float('nan'), 4.0, 5.0])
# Mittelwert der Nicht-NaN-Werte berechnen
mean = torch.nanmean(x)
print("Nicht-NaN-Mittelwert:", mean)
Das Ausgabeergebnis ist:
非 NaN 均值: tensor(3.)
Andere Erweiterungen