Comment puis-je extraire les filtres appris dans un calque Convolution?

Question

J'ai une couche Convolution dans mon code R, créé comme:
conv1 <- mx.symbol.Convolution(data=data, kernel=c(10,1), num_filter=10)

Une fois que le filet est complètement formé, comment puis-je extraire les 10 filtres?

Answer 1

Les poids de filtre se trouvent dans le paramètre weight de la convolution. En supposant que vous ayez utilisé la disposition standard, comme dans votre exemple, les poids auront une forme (num_filter, channels, kernel [0], kernel [1]).

Par exemple

conv1.weight.data()[0] 

accède au tenseur de poids du premier filtre dans le contexte actuel.