I implement permute plugin,
Dims getOutputDimensions(int index, const Dims* inputs, int nbInputDims) override
{
assert(nbInputDims == 1);
assert(index == 0);
assert(inputs[index].nbDims == 3);
Quadruple tmpOutputDims = {0};
for(int i=0;i<4;++i){
int tmp = mPermuteOrder.data[i] - 1;
if(tmp == -1){
tmpOutputDims.data[i] = 1;
}else{
tmpOutputDims.data[i] = inputs[0].d[tmp];
}
}
return DimsCHW(tmpOutputDims.data[1],tmpOutputDims.data[2],tmpOutputDims.data[3]); // #1
}
last row,return DimsNCHW(tmpOutputDims.data[0],tmpOutputDims.data[1],tmpOutputDims.data[2],tmpOutputDims.data[3]); // #2
Can I use #2 ?