Función PyTorch torch.msort
Manual de referencia de Pytorch torch
torch.msortEs una función en PyTorch que se utiliza para devolver un tensor ordenado. Devuelve una copia del tensor de entrada ordenada de forma ascendente, similar a aplanar (flatten) el tensor y luego aplicar reshape.
Definición de la función
torch.msort(input)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor
x = torch.tensor([3, 1, 2])
# Devolver el tensor ordenado
result = torch.msort(x)
print(result)
# Crear tensor
x = torch.tensor([3, 1, 2])
# Devolver el tensor ordenado
result = torch.msort(x)
print(result)
El resultado es:
tensor([1, 2, 3])
Otras extensiones