Tenseurs partiellement empilables
J'ai une sortie d'une couche convolutive dans PyTorch de la forme [b,c,h]où bse trouvent mes lots, cmes canaux et hles 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