Tenseurs partiellement empilables
J'ai une sortie d'une couche convolutive dans PyTorch de la forme [b,c,h]
où b
se trouvent mes lots, c
mes canaux et h
les fonctionnalités. Je veux les empiler pour les alimenter dans une couche entièrement connectée sans changer les lots, afin qu'ils aient la forme [b, c*h]
. Comment puis-je faire ceci?
Solution du problème
Il semble qu'un simple remodelage ou une vue devrait fonctionner :
input.shape # [b,c,h]
reshaped = input.view(input.shape[0],-1) # [b,c*h]
Commentaires
Enregistrer un commentaire