Fonction torch.roll de PyTorch

Pytorch torch 参考手册 Manuel de référence PyTorch torch

torch.rollIl s'agit d'une fonction de PyTorch utilisée pour décaler circulairement les éléments d'un tenseur.

Définition de fonction

torch.roll(input, shifts, dims)

Exemple d'utilisation

Exemple

import torch

x = torch.arange(8)

# Décaler vers l'avant de 2 positions
print(torch.roll(x, 2))

# Décaler vers l'arrière de 2 positions
print(torch.roll(x, -2))

Le résultat de sortie est:

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

Pytorch torch 参考手册 Manuel de référence PyTorch torch

Autres extensions