torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-AI-IOT

Seems like trt module will decrease input data`s dimension to 4D.

Open Fuzhiyuan opened this issue 3 years ago • 1 comments

Fuzhiyuan avatar Jun 17 '22 09:06 Fuzhiyuan

As I pass a 5d tensor into trt module 【torch2trt(module, input)】. And my project will raise an error, the input tensor dimension is 4D. I pass a 1x1x128x128x128, and in that time the shape is 320x8x8x8. What happened?

Fuzhiyuan avatar Jun 17 '22 09:06 Fuzhiyuan