Python tf.summary.graph用法及代碼示例

編寫 TensorFlow 圖摘要。

用法

tf.summary.graph(
    graph_data
)

僅在即刻模式下編寫tf.Graph 或tf.compat.v1.GraphDef 的實例作為摘要。在使用 tf.function 時，請優先使用跟蹤 API(tf.summary.trace_on、tf.summary.trace_off 和 tf.summary.trace_export)，它可以自動收集和記錄執行中的圖表。

使用示例：

writer = tf.summary.create_file_writer("/tmp/mylogs")

@tf.function
def f():
  x = constant_op.constant(2)
  y = constant_op.constant(3)
  return x**y

with writer.as_default():
  tf.summary.graph(f.get_concrete_function().graph)

# Another example:in a very rare use case, when you are dealing with a TF v1
# graph.
graph = tf.Graph()
with graph.as_default():
  c = tf.constant(30.0)
with writer.as_default():
  tf.summary.graph(graph)