Función torch.unsqueeze de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.unsqueezeEs una función en PyTorch que se utiliza para insertar una dimensión de tamaño 1 en una posición específica.

Definición de la función

torch.unsqueeze(input, dim)

Ejemplo de uso

Ejemplo

import torch

x = torch.randn(2, 3)
print("Forma original:", x.shape)

# Insertar dimensión en dim=0
y = torch.unsqueeze(x, dim=0)
print("Después de expandir dim=0:", y.shape)

# Insertar dimensión en dim=1
z = torch.unsqueeze(x, dim=1)
print("Después de expandir dim=1:", z.shape)

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones