PyTorch torch.flatten-Funktion
PyTorch torch Referenzhandbuch
torch.flattenist eine Funktion in PyTorch, mit der Tensoren abgeflacht werden.
Funktionsdefinition
torch.flatten(input, start_dim, end_dim, out)
Anwendungsbeispiel
Beispiel
import torch
x = torch.randn(2, 3, 4)
# Auf eine Dimension abflachen
y = torch.flatten(x)
print("Ursprüngliche Form:", x.shape)
print("Nach dem Abflachen:", y.shape)
# Abflachen ab einer Dimension
z = torch.flatten(x, start_dim=1)
print("Abflachen ab dim=1:", z.shape)
x = torch.randn(2, 3, 4)
# Auf eine Dimension abflachen
y = torch.flatten(x)
print("Ursprüngliche Form:", x.shape)
print("Nach dem Abflachen:", y.shape)
# Abflachen ab einer Dimension
z = torch.flatten(x, start_dim=1)
print("Abflachen ab dim=1:", z.shape)
Das Ausgabeergebnis ist:
原始形状: torch.Size([2, 3, 4]) 展平后: torch.Size([24]) 从 dim=1 展平: torch.Size([2, 12])
Andere Erweiterungen