Функция torch.roll в PyTorch

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

torch.rollЭто функция в PyTorch, используемая для циклического сдвига элементов тензора.

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

torch.roll(input, shifts, dims)

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

Пример

import torch

x = torch.arange(8)

# Сдвиг вперёд на 2 позиции
print(torch.roll(x, 2))

# Сдвиг назад на 2 позиции
print(torch.roll(x, -2))

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

tensor([6, 7, 0, 1, 2, 3, 4, 5])
tensor([2, 3, 4, 5, 6, 7, 0, 1])

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

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