PyTorch torch.unsqueeze-Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.unsqueezeist eine Funktion in PyTorch, die verwendet wird, um eine Dimension der Größe 1 an der angegebenen Position einzufügen.

Funktionsdefinition

torch.unsqueeze(input, dim)

Anwendungsbeispiel

Beispiel

import torch

x = torch.randn(2, 3)
print("Ursprüngliche Form:", x.shape)

# Dimension bei dim=0 einfügen
y = torch.unsqueeze(x, dim=0)
print("Nach dim=0-Erweiterung:", y.shape)

# Dimension bei dim=1 einfügen
z = torch.unsqueeze(x, dim=1)
print("Nach dim=1-Erweiterung:", z.shape)

Die Ausgabe ist:

原始形状: torch.Size([2, 3])
dim=0 扩展后: torch.Size([1, 2, 3])
dim=1 扩展后: torch.Size([2, 1, 3])

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen