Search code examples
pythonc++nvidiatensorrt

What is c++ version of tensorrt.volume?


What is c++ alternative for this python API call:

...
import tensorrt as trt
size = trt.volume(input) // how to do same in c++?
...

UPDATE nvidia has C++ api, it has alternatives for all python tensorrt API calls, except this one: https://docs.nvidia.com/deeplearning/sdk/tensorrt-api


Solution

  • Finally, I've found it inside TensorRT c++ samples:

    inline int64_t volume(const nvinfer1::Dims& d)
    {
        return std::accumulate(d.d, d.d + d.nbDims, 1, std::multiplies<int64_t>());
    }