I have 2 questions with the dimension of the tensor.


How do I convert my model output torch.Size([5, 1024, 7, 2, 4]) to torch.Size([5,1024,7,1,1]) ? I thought of slicing, but is that good idea? or there is another good way to do it?


I have two torch.Size([5,1024,7,1,1]) output from the network, I would like to concatenate these two outputs and feed to two-layer Fully Connected layer. (layer 1 output :1024, layer 2 output: 1). What would be the best way to do it?


Thank you.