weixin_39670246
weixin_39670246
2020-12-07 12:09

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

  • 点赞
  • 回答
  • 收藏
  • 复制链接分享

4条回答

为你推荐

换一换