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

Comment signer la connexion OKEx API version 5 avec websockets ?

La fonction GCP Cloud pour écrire des données dans BigQuery s'exécute avec succès, mais les données n'apparaissent pas dans la table BigQuery

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