Fonction torch.unsqueeze de PyTorch

Pytorch torch 参考手册 Manuel de référence PyTorch torch

torch.unsqueezeest une fonction de PyTorch utilisée pour insérer une dimension de taille 1 à la position spécifiée.

Définition de fonction

torch.unsqueeze(input, dim)

Exemple d'utilisation

Exemple

import torch

x = torch.randn(2, 3)
print("Forme d'origine :", x.shape)

# Insérer une dimension à dim=0
y = torch.unsqueeze(x, dim=0)
print("Après expansion dim=0 :", y.shape)

# Insérer une dimension à dim=1
z = torch.unsqueeze(x, dim=1)
print("Après expansion dim=1 :", z.shape)

La sortie est :

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

Pytorch torch 参考手册 Manuel de référence PyTorch torch

Autres extensions