PyTorch torch.nanmedian Funktion
PyTorch torch-Referenzhandbuch
torch.nanmedianEs ist eine Funktion in PyTorch, die verwendet wird, um den Median der Nicht-NaN-Werte in einem Tensor zurückzugeben. Im Gegensatz zu median ignoriert sie alle NaN-Werte.
Funktionsdefinition
torch.nanmedian(input, dim, keepdim=False)
Anwendungsbeispiel
Beispiel
import torch
x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])
# Median der Nicht-NaN-Werte zurückgeben
print("Nicht-NaN-Median:", torch.nanmedian(x))
# Nicht-NaN-Median entlang dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("dim=0 Nicht-NaN-Median:", torch.nanmedian(y, dim=0))
x = torch.tensor([1.0, float('nan'), 3.0, 2.0, 5.0])
# Median der Nicht-NaN-Werte zurückgeben
print("Nicht-NaN-Median:", torch.nanmedian(x))
# Nicht-NaN-Median entlang dim=0
y = torch.tensor([[1.0, float('nan'), 2.0], [4.0, 1.0, 3.0]])
print("dim=0 Nicht-NaN-Median:", torch.nanmedian(y, dim=0))
Das Ausgabeergebnis ist:
非NaN中位数: tensor(2.) dim=0 非NaN中位数: torch.return_types.nanmedian(values=tensor([1., 1., 2.]), indices=tensor([0, 1, 0]))
Andere Erweiterungen