Fonction torch.msort de PyTorch

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

torch.msortIl s'agit d'une fonction de PyTorch utilisée pour renvoyer un tenseur trié. Elle renvoie une copie du tenseur d'entrée organisée par ordre croissant, semblable au tri du tenseur après aplatissement puis redimensionnement.

Définition de fonction

torch.msort(input)

Exemple d'utilisation

Exemple

import torch

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

# Retourner le tenseur trié
result = torch.msort(x)
print(result)

Le résultat de sortie est:

tensor([1, 2, 3])

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

Autres extensions