PyTorch torch.roll Funktion
PyTorch torch-Referenzhandbuch
torch.rollDies ist eine Funktion in PyTorch, die zum Rollen von Tensor-Elementen verwendet wird.
Funktionsdefinition
torch.roll(input, shifts, dims)
Anwendungsbeispiel
Beispiel
import torch
x = torch.arange(8)
# Um 2 Positionen vorwärts rollen
print(torch.roll(x, 2))
# Um 2 Positionen rückwärts rollen
print(torch.roll(x, -2))
x = torch.arange(8)
# Um 2 Positionen vorwärts rollen
print(torch.roll(x, 2))
# Um 2 Positionen rückwärts rollen
print(torch.roll(x, -2))
Das Ausgabeergebnis ist:
tensor([6, 7, 0, 1, 2, 3, 4, 5]) tensor([2, 3, 4, 5, 6, 7, 0, 1])
Andere Erweiterungen