阅读(7.1k) 书签 (0)

TensorFlow函数教程:tf.io.write_graph

2019-02-27 17:27 更新

tf.io.write_graph函数

别名:

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

定义在:tensorflow/python/framework/graph_io.py。

将图形原型写入文件。

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

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

或者:

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

参数:

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

返回:

输出原型文件的路径。