Connect fully connected layer to a convolution layer?

Question

3 个投票

When building a CNN network for image reconstruction with fully connected layer followed by convolution layers, I found that the output of fully connected layer is always a vector. If I want fully connected layer to be followed by a 2D convolution layer, is there any intermediate layer to reshape the data transferred in between two layers, either on output of fully connected layer or input of convolution layers?

Thanks.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Follow Question

Answer 1

Uttiya Ghosh 2020-6-18

0 个投票

Hi Kittel,

From my understanding, you want to introduce a reshape layer between your fully connected layer and 2D convolution layer. There is no reshape layer in MATLAB. As a workaround you can always create a custom layer.

For more information, refer to the following links.

https://www.mathworks.com/matlabcentral/answers/461988-reshape-layer-in-deep-learning#accepted_answer_376083

https://www.mathworks.com/help/deeplearning/ug/list-of-deep-learning-layers.html