Hi All,
I have a 4 dimensional numpy input array that was created from extracting features using ResNet50.
With a batch size of 5, I have created tensors of:
torch.Size([5, 2048, 1, 1])

I’m not really sure what to do if I want to feed this into a simple autoencoder. Would anyone be able to clarify how I should reshape this tensor to feed into an nn.linear layer?
Cheers,