Fonction torch.flatten de PyTorch

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

torch.flattenest une fonction de PyTorch utilisée pour aplatir les tenseurs.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

# Aplatir en une dimension
y = torch.flatten(x)

print("Forme d'origine :", x.shape)
print("Après aplatissement :", y.shape)

# Aplatir à partir d'une dimension spécifiée
z = torch.flatten(x, start_dim=1)
print("Aplatir à partir de dim=1 :", z.shape)

Le résultat de sortie est:

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

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

Autres extensions