Solved by KFrank in post #2 
              
              
                Hi Taran! 

I assume that your problem is the last two length-1 dimensions 
of your tensor.  If so, you can use torch.squeeze() to get rid of 
them. 
You could then feed your (now torch.Size([5, 2048])) 
tensor to, for example, a nn.Linear (2048, 256) layer. 
(Note, if you ever have a batch size of …