What if I have something like this:
conv1 -> flatten -> dense -> dense -> reshape -> conv2
Where conv1’s output shape is (batch, c, h, w) and conv2’s input shape is (batch, h, w, c). Do you think I should permute dimension somewhere in between, or do you think the fully connected layers will do the mapping?