Fonction torch.argsort de PyTorch

Pytorch torch 参考手册 Manuel de référence PyTorch torch

torch.argsortC'est une fonction dans PyTorch utilisée pour retourner les indices après le tri. Elle retourne les indices de position d'origine de chaque élément après le tri du tenseur.

Définition de fonction

torch.argsort(input, dim=-1, descending=False, stable=True)

Exemples d'utilisation

Exemple

import torch

# Créer un tenseur
x = torch.tensor([3, 1, 2])

# Retourner les indices triés
indices = torch.argsort(x)
print(indices)

Le résultat de sortie est:

tensor([1, 2, 0])

Pytorch torch 参考手册 Manuel de référence PyTorch torch

Autres extensions