Функция PyTorch torch.argsort

Pytorch torch 参考手册 Справочное руководство PyTorch torch

torch.argsortЭто функция в PyTorch, используемая для возврата индексов после сортировки. Она возвращает индексы исходных позиций каждого элемента после сортировки тензора.

Определение функции

torch.argsort(input, dim=-1, descending=False, stable=True)

Примеры использования

Пример

import torch

# Создать тензор
x = torch.tensor([3, 1, 2])

# Возвращает отсортированные индексы
indices = torch.argsort(x)
print(indices)

Результат вывода:

tensor([1, 2, 0])

Pytorch torch 参考手册 Справочное руководство PyTorch torch

Другие расширения