当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.io.write_graph用法及代码示例


将图形原型写入文件。

用法

tf.io.write_graph(
    graph_or_graph_def, logdir, name, as_text=True
)

参数

  • graph_or_graph_def Graph 或 GraphDef 协议缓冲区。
  • logdir 写入图表的目录。这可以指远程文件系统，例如 Google Cloud Storage (GCS)。
  • name 图表的文件名。
  • as_text 如果 True ，将图形写入 ASCII 原型。

返回

  • 输出 proto 文件的路径。

除非 as_text 是 False ，否则该图将被编写为文本原型。

v = tf.Variable(0, name='my_variable')
sess = tf.compat.v1.Session()
tf.io.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt')

或者

v = tf.Variable(0, name='my_variable')
sess = tf.compat.v1.Session()
tf.io.write_graph(sess.graph, '/tmp/my-model', 'train.pbtxt')

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.io.write_graph。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。