getOutputDimensions for multiple outputs in plugin

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

In plugin creation, we need to return Output Dims.

    Dims getOutputDimensions(int index, const Dims* inputs, int nbInputDims) override
    {
        assert(index == 0 && nbInputDims == 2 && inputs[0].nbDims == 3 && inputs[1].nbDims == 1);
       
        return Dims3(inputs[0].d[0], outDims.h(), outDims.w());
    }

If my plugin has two outputs, how should I return?

2

Hi @edit_or
The below post should be able to answer your queries.

Thanks.