MATLAB: Reshape layer in deep learning

Question

I want to try out a deep learning architecture which first does fully connected layers and then transitions into image convolutions. So after the fully connected layers I want to reshape the weights from a vector into a image like matrix. Is there a reshape layer within the Deep learning toolbox which does this?

Answer 1

There is no reshape layer in MATLAB which changes output from fully connected layer into image like matrix. You can create a custom deep learning layer to solve this problem. You can refer to this documentation link for more details on creating a custom layer: https://www.mathworks.com/help/deeplearning/ug/define-custom-deep-learning-layers.html

Hope this helps.