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

Question

這個python API調用的c++替代品是什么：

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

更新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>());
}