PyTorch torch.permute-Funktion
PyTorch torch Referenzhandbuch
torch.permuteist eine Funktion in PyTorch, die verwendet wird, um Tensor-Dimensionen neu anzuordnen.
Funktionsdefinition
torch.permute(input, dims)
Anwendungsbeispiele
Beispiel
import torch
# Erstelle einen 3D-Tensor
x = torch.randn(2, 3, 4)
print("Ursprüngliche Form:", x.shape)
# Dimensionen neu anordnen (2,3,4) -> (4,2,3)
y = x.permute(2, 0, 1)
print("Nach permute:", y.shape)
# Erstelle einen 3D-Tensor
x = torch.randn(2, 3, 4)
print("Ursprüngliche Form:", x.shape)
# Dimensionen neu anordnen (2,3,4) -> (4,2,3)
y = x.permute(2, 0, 1)
print("Nach permute:", y.shape)
Andere Erweiterungen