Fonction torch.sort de PyTorch
Manuel de référence PyTorch torch
torch.sortest une fonction dans PyTorch utilisée pour trier selon une dimension spécifiée.
Définition de fonction
torch.sort(input, dim, descending, stable, out)
Exemple d'utilisation
Exemple
import torch
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Trier
values, indices = torch.sort(x)
print("Valeurs triées :")
print(values)
print("Indices triés :")
print(indices)
# Trier par ordre décroissant
values_desc, _ = torch.sort(x, descending=True)
print("Ordre décroissant :")
print(values_desc)
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Trier
values, indices = torch.sort(x)
print("Valeurs triées :")
print(values)
print("Indices triés :")
print(indices)
# Trier par ordre décroissant
values_desc, _ = torch.sort(x, descending=True)
print("Ordre décroissant :")
print(values_desc)
Autres extensions