标签: c++ tensorflow
我将TensorFlow包装在一个外部工具中,该工具需要在运行图形之前知道输出的大小。 {C}中从磁盘加载tensorflow::GraphDef。
tensorflow::GraphDef
我可以运行图表进行一次迭代并读取输出形状,但这可能相当于只是为了读取形状而进行的大量处理。
可以想象将输入馈送到图形,仅将形状信息传播到输出节点并从那里读取它,但我无法找到这样做的方法。
是否可以在不运行TensorFlow图的情况下使用TensorFlow C ++ API进行形状推断?