Comment puis-je utiliser des calques de tableau au chalumeau pour créer mon propre réseau?

Question

Je suis coincé sur nn package dans torch7, en particulier ses couches de table. Comment puis-je utiliser son ConcatTable, CAddTable ou d'autres méthodes pour créer un réseau ci-dessous?

Ici, j'ai deux branche de mon réseau, l'un contient 3 couches de convolution et une autre une seule couche. Je voulais additionner les sorties de mes deux dernières couches de convolution (somme de la sortie de la convolution 4ème et de la 5ème convolution), comment écrire mon code de torche en utilisant le paquet nn.

Answer 1

En supposant que votre branches sont correctement mises en œuvre:

local net = nn.Sequential() 
    :add(conv1) 
    :add(nn.ConcatTable() 
     :add(branch1) 
     :add(branch2)) 
    :add(nn.CAddTable()) 

nn.CAddTable est à ne pas confondre avec nn.Sum. Le premier reçoit une table de tenseurs et retourne la sommation de tout tenseur tandis que le second reçoit un seul tenseur et calcule la somme de ses éléments selon la dimension spécifiée.