How to Connect Convolutional layer to Fully Connected layer in Pytorch while Implementing SRGAN
You simply reshape the tensor to (batch_size, n_nodes) using tensor.view(). In your specific case this would be x.view(x.size()[0], -1).