input layer and flatten mismatch

 採用された回答

Abolfazl Chaman Motlagh
Abolfazl Chaman Motlagh 2021 年 12 月 12 日

0 投票

The Flatten layer support sequence input only. [1]
As i understand from your network architechture you don't need Flatten Layer. in Matlab Flattenlayer is for merging height,width and channel of input in an aray for N data in minibatch and S (number of sequences) series data. the output is a 3-dimensianal array with size of (H*W*C)xNxS.
if you use a fullyconnectedlayer(fc) as you did after Flatten layer, the input array of fc layer which is comming from a Conv layer become vectorize in a spatial domain in output of layer.
so just deleting the Flatten layer will solve the problem.

その他の回答 (0 件)

カテゴリ

ヘルプ センター および File ExchangeDeep Learning Toolbox についてさらに検索

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by