permute for 3 dim tensor

Hi jaybdub,

If I got a tensor which has x.shape = [1,1024,22], and I want to permute the dims with x = x.permute(2,0,1) but it went wrong(AssertionError).What should I do to get the same result and the process can be transferred to tensorrt?

该提问来源于开源项目：NVIDIA-AI-IOT/torch2trt