PyTorch torch.rot90 Funktion
PyTorch torch Referenzhandbuch
torch.rot90Es ist eine Funktion in PyTorch, mit der Tensoren um 90 Grad gedreht werden.
Funktionsdefinition
torch.rot90(input, k=1, dims=[0, 1])
Parameterbeschreibung:
input- Eingabe-Tensork- Anzahl der Drehungen; positiv bedeutet gegen den Uhrzeigersinn, negativ bedeutet im Uhrzeigersinn (Standard 1)dims- Ebenendimensionen für die Drehung (Standard [0, 1])
Anwendungsbeispiele
Beispiel
import torch
# 2D-Tensor-Drehung
x = torch.arange(12).reshape(3, 4)
print("Original-Tensor:")
print(x)
result = torch.rot90(x, 1)
print("Drehung um 90 Grad gegen den Uhrzeigersinn:")
print(result)
result = torch.rot90(x, -1)
print("nDrehung um 90 Grad im Uhrzeigersinn:")
print(result)
result = torch.rot90(x, 2)
print("nDrehung um 180 Grad:")
print(result)
# Drehen entlang einer angegebenen Ebene (3D-Tensor)
y = torch.arange(24).reshape(2, 3, 4)
print("nForm des 3D-Tensors:", y.shape)
result = torch.rot90(y, 1, dims=[1, 2])
print("Form nach Drehung um 90 Grad entlang der [1, 2]-Ebene:", result.shape)
# 2D-Tensor-Drehung
x = torch.arange(12).reshape(3, 4)
print("Original-Tensor:")
print(x)
result = torch.rot90(x, 1)
print("Drehung um 90 Grad gegen den Uhrzeigersinn:")
print(result)
result = torch.rot90(x, -1)
print("nDrehung um 90 Grad im Uhrzeigersinn:")
print(result)
result = torch.rot90(x, 2)
print("nDrehung um 180 Grad:")
print(result)
# Drehen entlang einer angegebenen Ebene (3D-Tensor)
y = torch.arange(24).reshape(2, 3, 4)
print("nForm des 3D-Tensors:", y.shape)
result = torch.rot90(y, 1, dims=[1, 2])
print("Form nach Drehung um 90 Grad entlang der [1, 2]-Ebene:", result.shape)
Das Ausgabeergebnis ist:
原始张量:
tensor([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]])
逆时针旋转 90 度:
tensor([[ 3, 7, 11],
[ 2, 6, 10],
[ 1, 5, 9],
[ 0, 4, 8]])
顺时针旋转 90 度:
tensor([[ 8, 4, 0],
[ 9, 5, 1],
[10, 6, 2],
[11, 7, 3]])
旋转 180 度:
tensor([[11, 10, 9, 8],
[ 7, 6, 5, 4],
[ 3, 2, 1, 0]])
三维张量形状: torch.Size([2, 3, 4])
沿 [1, 2] 平面旋转 90 度后形状: torch.Size([2, 4, 3])
Andere Erweiterungen