PyTorch torch.argsort-Funktion
PyTorch torch-Referenzhandbuch
torch.argsortEs ist eine Funktion in PyTorch, die verwendet wird, um die Indizes nach dem Sortieren zurückzugeben. Sie gibt die ursprünglichen Positionsindizes jedes Elements zurück, nachdem der Tensor sortiert wurde.
Funktionsdefinition
torch.argsort(input, dim=-1, descending=False, stable=True)
Anwendungsbeispiele
Beispiel
import torch
# Tensor erstellen
x = torch.tensor([3, 1, 2])
# Sortierte Indizes zurückgeben
indices = torch.argsort(x)
print(indices)
# Tensor erstellen
x = torch.tensor([3, 1, 2])
# Sortierte Indizes zurückgeben
indices = torch.argsort(x)
print(indices)
Das Ausgabeergebnis ist:
tensor([1, 2, 0])
Andere Erweiterungen