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

时间:2018-05-23 13:37:23

标签: c++ tensorflow

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

我可以运行图表进行一次迭代并读取输出形状,但这可能相当于只是为了读取形状而进行的大量处理。

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

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

0 个答案:

没有答案