TensorFlow函數(shù)教程：tf.io.write_graph

tf.io.write_graph函數(shù)

別名：

• 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。

將圖形原型寫(xiě)入文件。

除非as_text是False，否則該圖形將被寫(xiě)為文本原型。

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')

參數(shù)：

• graph_or_graph_def：Graph或GraphDef協(xié)議緩沖區(qū)。
• logdir：寫(xiě)入圖形的目錄。這可以指遠(yuǎn)程文件系統(tǒng)，例如Google Cloud Storage（GCS）。
• name：圖形的文件名。
• as_text：如果為T(mén)rue，將圖形寫(xiě)為ASCII原型。

返回：

輸出原型文件的路徑。