如何用C ++ API解析tensorflow模型

时间:2017-03-09 14:45:16

标签: tensorflow deep-learning

我想解析预先训练的张量流模型。例如,我想获得操作节点的完整列表,包括给定模型的名称和依赖关系。

所以,首先我搜索了Java API,显然Java接口支持的API很少。所以我寻求C ++ API,但未能找到合适的API。

我不使用python的原因是我需要在Android设备上执行此操作。

1 个答案:

答案 0 :(得分:1)

TensorFlow图存储为GraphDef协议缓冲区。您应该能够构建它的java版本并使用它来检查存储的图形。这将包含操作列表及其依赖项,但将具有权重值。