How to do reshape in a NN?

Question

Ma Yuelin 2018-10-29

2
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/426646-how-to-do-reshape-in-a-nn

回答： Rahul 2025-4-7

How can I reshape a NN's layer's output? I have a fully connected layer whose output is a 1D vector. I want to reshape it to a 3D matrix and apply convolution layer after it. Say, is there a similar function or something like "reshape layer" in Tensorflow?

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

请先登录，再进行评论。

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

Answer 1

Rahul 2025-4-7

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/426646-how-to-do-reshape-in-a-nn#answer_1563251

Hi @Ma Yuelin,

There are multiple ways to approach the issue in MATLAB. The following can be considered:

'Deep Network Designer' app can be used to adjust layers according to custom requirement.
Since there is no direct 'reshapeLayer' function avaialble, a similar functionality can be acheived using 'funcionLayer' function. This is also mentioned in this MATLAB Answer: https://www.mathworks.com/matlabcentral/answers/461988-reshape-layer-in-deep-learning. The section of 'Reformat Data Using Function Layer' would provide necessary information.
Defining a custom class for the output layer, where the 'reshape' function can be used to adjust the shape of the output layer.

The following Mathworks documentations can be referred to know more:

'Deep Network Designer': https://www.mathworks.com/help/deeplearning/ref/deepnetworkdesigner-app.html

'functionLayer': https://www.mathworks.com/help/deeplearning/ref/nnet.cnn.layer.functionlayer.html

'reshape': https://www.mathworks.com/help/matlab/ref/double.reshape.html

'Defining custom Deep Learning Layer': https://www.mathworks.com/help/deeplearning/ug/define-custom-deep-learning-layer-with-formatted-inputs.html

Thanks.