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?
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.