I am trying to build a convolutional neural network to classify different boat types. I have input images of dimensions [3, 512, 512]. I am unsure as to how to feed the output of convolutional layers into a linear layer in my network.