In my project, I use a permute layer in caffe prototxt. I wanted to permute the shape NCHW (batchchannelheightwidth) to WNC*H, but when I transformed this caffe model to tensorrt model, the nvcaffeparser1::ICaffeParser::parse() function throw the following error:
TensorRT does not support permute in N (batch) dimension, and order index must be within the tensor dimensions.
Could you tell me how to solve this problem? Should I implement a new layer in tensorrt using the IPlugin layer or are there any other solution in tensorrt?