Función torch.flatten de PyTorch

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

torch.flattenEs una función en PyTorch utilizada para aplanar tensores.

Definición de la función

torch.flatten(input, start_dim, end_dim, out)

Ejemplo de uso

Ejemplo

import torch

x = torch.randn(2, 3, 4)

# Aplanar a una dimensión
y = torch.flatten(x)

print("Forma original:", x.shape)
print("Después de aplanar:", y.shape)

# Aplanar desde la dimensión especificada
z = torch.flatten(x, start_dim=1)
print("Aplanar desde dim=1:", z.shape)

El resultado de salida es:

原始形状: torch.Size([2, 3, 4])
展平后: torch.Size([24])
从 dim=1 展平: torch.Size([2, 12])

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

Otras extensiones