I’m trying to change the output batch dimension.
ie) [Input] 236 → [output] 6*8
But I found that the tensorrt library can’t change the output batch dimension.
Even when I’m using nvinfer1::DimsNCHW({6,8,1,1}) in getOutputDimention() function,
I got 506811 shaped tensor, which is the result of adding the batch size of previous node.
Why does not the batch size change?