PyTorch torch.squeeze-Funktion
PyTorch torch-Referenzhandbuch
torch.squeezeEs ist eine Funktion in PyTorch, die verwendet wird, um Dimensionen der Größe 1 aus einem Tensor zu entfernen.
Funktionsdefinition
torch.squeeze(input, dim)
Anwendungsbeispiel
Beispiel
import torch
# Erstelle einen Tensor mit Dimensionen der Größe 1
x = torch.randn(1, 2, 1, 3, 1)
print("Ursprüngliche Form:", x.shape)
# Entferne alle Dimensionen der Größe 1
y = torch.squeeze(x)
print("Form nach Squeeze:", y.shape)
# Dimension angeben
z = torch.squeeze(x, dim=0)
print("Nach Squeeze mit dim=0:", z.shape)
# Erstelle einen Tensor mit Dimensionen der Größe 1
x = torch.randn(1, 2, 1, 3, 1)
print("Ursprüngliche Form:", x.shape)
# Entferne alle Dimensionen der Größe 1
y = torch.squeeze(x)
print("Form nach Squeeze:", y.shape)
# Dimension angeben
z = torch.squeeze(x, dim=0)
print("Nach Squeeze mit dim=0:", z.shape)
Die Ausgabe ist:
原始形状: torch.Size([1, 2, 1, 3, 1]) 挤压后形状: torch.Size([2, 3]) dim=0 挤压后: torch.Size([2, 1, 3, 1])
Andere Erweiterungen