Test it on the Peltarion Platform
A platform to build and deploy deep learning projects.
Even if you’re not an AI superstar.
The Flatten block reshapes the input, which could be a tensor of any shape, into a 1-dimensional vector.
One of its uses is to feed data coming from convolutional layers into a Dense layer, for instance in the middle of some auto-encoders. It is also useful when wanting to Concatenate a convolutional part of a model with an MLP part that has Dense layers, since Dense layers operate on 1D input.