tensorrt.volume 的 C++ 版本是什么？

Question

此 python API 调用的 c++ 替代方案是什么：

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

UPDATE nvidia 有 C++ api，它有所有 python tensorrt API 调用的替代方案，除了这个：https://docs.nvidia.com/deeplearning/sdk/tensorrt-api

Answer 1

最后，我在 TensorRT c++ 示例中找到了它：

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