tensorflow 源代码学习之图(graph_def 和 graph）

最新推荐文章于 2023-11-07 20:38:28 发布

张悠悠爸

最新推荐文章于 2023-11-07 20:38:28 发布

阅读量1.3k

点赞数

分类专栏： Tensorflow 文章标签： tensorflow

本文链接：https://blog.csdn.net/challno/article/details/54600494

版权

本文深入探讨TensorFlow中的图构建过程，包括从字符串转换为GraphDef对象，再进一步转化为Graph对象。文章指出，在创建Graph时会默认添加_SOURCE和_SINK节点，解释了图实质上是节点及其操作和边的关系。

摘要由CSDN通过智能技术生成

tensorflow 源代码学习之图(graph_def 和 graph）

注：辉辉 2017/1/18 12:48:46
/tensorflow/core/framework/graph.proto文件是GraphDef的原型声明，对应生成graph.pb.h 和graph.pb.cc文件。（protocol buffer compiler）
/tensorflow/core/framework/node_def.proto文件是NodeDef的原型声明，对应生成node_def.pb.h和graph.pb.cc文件。（protocol buffer compiler）

一. 构建Graph

输入以字符串形式提供自定义图(按照graph.proto原型)：

const string  my_graph_def_ascii =
                "node { name: 'A' op: 'TestInput' }"
                "node { name: 'B' op: 'TestParams' }"
                "node { name: 'C' op: 'TestMul' input: [ 'A:0', 'B'] }"
                "node { name: 'D' op: 'TestMul' input: [ 'A:1', 'C'] }";