I have a tensor output that is size torch.Size([1, 6, 1])  that I want to have the final shape of torch.Size([6]). I tried doing output.view(-1,6) but that gives me a tensor of size torch.Size([1, 6]). Could someone help me?