How do I recover the 3D structure of a layer after a Fully Connected layer (or a flatten layer)?

I want to apply 1 fully connected layer first and then start apply convolutions. But I noticed that the 3D structure might be lost. Is there a way to recover that 3D structure and apply 2d convolutions or am I doomed to only apply 1d convolutions thereafter?


Cross posted: