@ptrblck  Sorry for specific tagging I will not do this again. I solved this problem.
I have another question I have a tensor of shape [1, 1, 64000] where the first 1 is the batch size can I make it of shape [1, 64000] i.e remove the first dim.