I convert tensorflow pb model to tensorrt plan model, and I find the output shape of pb model([1,4]) mismatch the converted tensorrt plan model([1,1,1,4]). I am not sure which step will expand the dims. The convert step show below:
- Use the command "convert-to-uff ***.pb " to obtain ***.uff.
- Use the tensorrt C++ function “loadModelAndCreateEngine()” to generate ***.plan stream.
I can see the output shape of plan is [1,1,1,4], but not sure which step convert [1,4] to [1,1,1,4].
Please help me. Thanks