Tenseurs partiellement empilables

J'ai une sortie d'une couche convolutive dans PyTorch de la forme [b,c,h]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

Posts les plus consultés de ce blog

Erreur Symfony : "Une exception a été levée lors du rendu d'un modèle"

Détecter les appuis sur les touches fléchées en JavaScript

Une chaîne vide donne "Des erreurs ont été détectées dans les arguments de la ligne de commande, veuillez vous assurer que tous les arguments sont correctement définis"