Функция PyTorch torch.sort
Справочное руководство PyTorch torch
torch.sort— это функция в PyTorch, используемая для сортировки по указанному измерению.
Определение функции
torch.sort(input, dim, descending, stable, out)
Пример использования
Пример
import torch
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Сортировка
values, indices = torch.sort(x)
print("Отсортированные значения:")
print(values)
print("Отсортированные индексы:")
print(indices)
# Сортировка по убыванию
values_desc, _ = torch.sort(x, descending=True)
print("По убыванию:")
print(values_desc)
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Сортировка
values, indices = torch.sort(x)
print("Отсортированные значения:")
print(values)
print("Отсортированные индексы:")
print(indices)
# Сортировка по убыванию
values_desc, _ = torch.sort(x, descending=True)
print("По убыванию:")
print(values_desc)
Другие расширения