什么是 c++ 版本的 tensorrt.volume？答案

【问题标题】：What is c++ version of tensorrt.volume?什么是 c++ 版本的 tensorrt.volume？
【发布时间】：2019-09-07 17:00:43
【问题描述】：

此 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

【问题讨论】：

@SamVarshavchik 除了这不是原生 Python 库。有许多项目提供多种语言的绑定，例如 OpenCV。这本质上是一个带有 Python 绑定的 C++ 库，我们一直在处理 Python 和 C++ 之间的 OpenCV 问题映射。看起来 TensorRT 是一个类似的项目。因为 Python 绑定是建立在 C++ API 之上的，所以很有可能。是替代品。
@SamVarshavchik 此外，即使不是这种情况，也有更具建设性的方法来解释切换到完全不同的编码环境的复杂性。

【解决方案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>());
}

【讨论】：