首页 > 解决方案 > 是否可以在不运行 TensorFlow 图的情况下使用 TensorFlow C++ API 进行形状推断?

问题描述

我将 TensorFlow 包装在一个外部工具中,该工具需要在图表运行之前知道输出的大小。是从 C++ 中的tensorflow::GraphDef磁盘加载的。

我可以运行图形进行一次迭代并读取输出形状,但这可能仅用于读取形状就需要进行大量处理。

可以想象将输入输入到图形中,仅将形状信息传播到输出节点并从那里读取它,但我一直无法找到这样做的方法。

是否可以在不运行 TensorFlow 图的情况下使用 TensorFlow C++ API 进行形状推断?

标签: c++tensorflow

解决方案


推荐阅读