Функция PyTorch torch.msort
Справочное руководство PyTorch torch
torch.msortЭто функция в PyTorch, используемая для возврата отсортированного тензора. Она возвращает копию входного тензора, упорядоченную по возрастанию, подобно сортировке тензора после выравнивания и последующего изменения формы.
Определение функции
torch.msort(input)
Пример использования
Пример
import torch
# Создать тензор
x = torch.tensor([3, 1, 2])
# Вернуть отсортированный тензор
result = torch.msort(x)
print(result)
# Создать тензор
x = torch.tensor([3, 1, 2])
# Вернуть отсортированный тензор
result = torch.msort(x)
print(result)
Результат вывода:
tensor([1, 2, 3])
Другие расширения