當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。