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)