PyTorch torch.median Funktion
PyTorch torch-Referenzhandbuch
torch.medianEs ist eine Funktion in PyTorch, die verwendet wird, um den Median eines Tensors zurückzugeben. Der Median ist der mittlere Wert nach dem Sortieren der Daten.
Funktionsdefinition
torch.median(input, dim, keepdim=False)
Anwendungsbeispiel
Beispiel
import torch
x = torch.tensor([4, 2, 1, 3, 5])
# Median aller Elemente zurückgeben
print("Globaler Median:", torch.median(x))
# Median entlang dim=0
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("Median für dim=0:", torch.median(y, dim=0))
x = torch.tensor([4, 2, 1, 3, 5])
# Median aller Elemente zurückgeben
print("Globaler Median:", torch.median(x))
# Median entlang dim=0
y = torch.tensor([[1, 3, 2], [4, 1, 3]])
print("Median für dim=0:", torch.median(y, dim=0))
Das Ausgabeergebnis ist:
全局中位数: tensor(3) dim=0 中位数: torch.return_types.median(values=tensor([1, 1, 2]), indices=tensor([0, 1, 0]))
Andere Erweiterungen