PyTorch torch.mode Funktion
PyTorch torch Referenzhandbuch
torch.modeEs ist eine Funktion in PyTorch, die verwendet wird, um den Modus eines Tensors zurückzugeben. Der Modus ist der Wert, der in einer Datenmenge am häufigsten vorkommt.
Funktionsdefinition
torch.mode(input, dim, keepdim=False)
Anwendungsbeispiel
Beispiel
import torch
x = torch.tensor([1, 2, 2, 3, 3, 3])
# Modus zurückgeben
print("Modus:", torch.mode(x))
# Modus entlang dim=0
y = torch.tensor([[1, 2, 2], [1, 2, 3]])
print("dim=0 Modus:", torch.mode(y, dim=0))
# Modus entlang dim=1
print("dim=1 Modus:", torch.mode(y, dim=1))
x = torch.tensor([1, 2, 2, 3, 3, 3])
# Modus zurückgeben
print("Modus:", torch.mode(x))
# Modus entlang dim=0
y = torch.tensor([[1, 2, 2], [1, 2, 3]])
print("dim=0 Modus:", torch.mode(y, dim=0))
# Modus entlang dim=1
print("dim=1 Modus:", torch.mode(y, dim=1))
Das Ausgabeergebnis ist:
众数: torch.return_types.mode(values=tensor(3), indices=tensor(4)) dim=0 众数: torch.return_types.mode(values=tensor([1, 2, 2]), indices=tensor([0, 0, 0])) dim=1 众数: torch.return_types.mode(values=tensor([2, 3]), indices=tensor([1, 2]))
Andere Erweiterungen