Función PyTorch torch.roll

Pytorch torch 参考手册 Manual de referencia de Pytorch torch

torch.rollEs una función en PyTorch para hacer roll de los elementos de un tensor.

Definición de la función

torch.roll(input, shifts, dims)

Ejemplo de uso

Ejemplo

import torch

x = torch.arange(8)

# Rodar hacia adelante 2 posiciones
print(torch.roll(x, 2))

# Rodar hacia atrás 2 posiciones
print(torch.roll(x, -2))

El resultado de salida es:

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

Pytorch torch 参考手册 Manual de referencia de Pytorch torch

Otras extensiones