Función torch.sort de PyTorch
Manual de referencia de PyTorch torch
torch.sortEs una función en PyTorch que se utiliza para ordenar a lo largo de una dimensión especificada.
Definición de la función
torch.sort(input, dim, descending, stable, out)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Ordenar
values, indices = torch.sort(x)
print("Valores ordenados:")
print(values)
print("Índices ordenados:")
print(indices)
# Orden descendente
values_desc, _ = torch.sort(x, descending=True)
print("Descendente:")
print(values_desc)
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Ordenar
values, indices = torch.sort(x)
print("Valores ordenados:")
print(values)
print("Índices ordenados:")
print(indices)
# Orden descendente
values_desc, _ = torch.sort(x, descending=True)
print("Descendente:")
print(values_desc)
Otras extensiones