Questions I have a tensor of shape [5, 2, 18, 4096]. I want to stack the 0th dimension along the 2nd dimension. How can I do it? byMRDecember 23, 2022