Función torch.argsort de PyTorch
Manual de referencia de torch de PyTorch
torch.argsortEs una función en PyTorch que se utiliza para devolver los índices después de ordenar. Devuelve los índices de las posiciones originales de cada elemento una vez ordenado el tensor.
Definición de la función
torch.argsort(input, dim=-1, descending=False, stable=True)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor
x = torch.tensor([3, 1, 2])
# Devolver los índices ordenados
indices = torch.argsort(x)
print(indices)
# Crear tensor
x = torch.tensor([3, 1, 2])
# Devolver los índices ordenados
indices = torch.argsort(x)
print(indices)
El resultado de salida es:
tensor([1, 2, 0])
Otras extensiones