Connect fully connected layer to a convolution layer?
17 views (last 30 days)
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?
Uttiya Ghosh on 18 Jun 2020
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.