nbInputDims in getOutputDimensions

What is nbInputDims in getOutputDimensions?

Dims getOutputDimensions(int index, const Dims* inputs, int nbInputDims) override{
}

Say I have two inputs.

input_0  (88, 1, 43)
input_1  (1,)

What should be in nbInputDims?
Is it 3 Or 2?

Hi @edit_or
Please check the below link.
https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-713/api/c_api/classnvinfer1_1_1_i_plugin_v2_dynamic_ext.html#ad4ed0a21d021ff864aaa162e1f38f502
Here nbInputDims is the number of input tensors.
So looks like it will be 2 in this case
Thanks!

Thanks. For me is 2.

1 Like