PyTorch torch.sort-Funktion
PyTorch torch-Referenzhandbuch
torch.sortist eine Funktion in PyTorch, die zum Sortieren entlang einer angegebenen Dimension verwendet wird.
Funktionsdefinition
torch.sort(input, dim, descending, stable, out)
Anwendungsbeispiel
Beispiel
import torch
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Sortieren
values, indices = torch.sort(x)
print("Sortierte Werte:")
print(values)
print("Sortierte Indizes:")
print(indices)
# In absteigender Reihenfolge sortieren
values_desc, _ = torch.sort(x, descending=True)
print("Absteigende Reihenfolge:")
print(values_desc)
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Sortieren
values, indices = torch.sort(x)
print("Sortierte Werte:")
print(values)
print("Sortierte Indizes:")
print(indices)
# In absteigender Reihenfolge sortieren
values_desc, _ = torch.sort(x, descending=True)
print("Absteigende Reihenfolge:")
print(values_desc)
Andere Erweiterungen