TensorFlow 2.0 Beta is available Learn more

Module: tf.compat.v2.summary

View source on GitHub

Operations for writing summary data, for use in analysis and visualization.

The tf.summary module provides APIs for writing summary data. This data can be visualized in TensorBoard, the visualization toolkit that comes with TensorFlow. See the TensorBoard website for more detailed tutorials about how to use these APIs, or some quick examples below.

Example usage with eager execution, the default in TF 2.0:

writer = tf.summary.create_file_writer("/tmp/mylogs")
with writer.as_default():
  for step in range(100):
    # other model code would go here
    tf.summary.scalar("my_metric", 0.5, step=step)
    writer.flush()

Example usage with tf.function graph execution:

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

@tf.function
def my_func(step):
  # other model code would go here
  with writer.as_default():
    tf.summary.scalar("my_metric", 0.5, step=step)

for step in range(100):
  my_func(step)
  writer.flush()

Example usage with legacy TF 1.x graph execution:

with tf.compat.v1.Graph().as_default():
  step = tf.Variable(0, dtype=tf.int64)
  step_update = step.assign_add(1)
  writer = tf.summary.create_file_writer("/tmp/mylogs")
  with writer.as_default():
    tf.summary.scalar("my_metric", 0.5, step=step)
  all_summary_ops = tf.compat.v1.summary.all_v2_summary_ops()
  writer_flush = writer.flush()

  sess = tf.compat.v1.Session()
  sess.run([writer.init(), step.initializer])
  for i in range(100):
    sess.run(all_summary_ops)
    sess.run(step_update)
    sess.run(writer_flush)

Modules

experimental module: Public API for tf.summary.experimental namespace.

Classes

class SummaryWriter: Interface representing a stateful summary writer object.

Functions

audio(...): Write an audio summary.

create_file_writer(...): Creates a summary file writer for the given log directory.

create_noop_writer(...): Returns a summary writer that does nothing.

flush(...): Forces summary writer to send any buffered data to storage.

histogram(...): Write a histogram summary.

image(...): Write an image summary.

record_if(...): Sets summary recording on or off per the provided boolean value.

scalar(...): Write a scalar summary.

text(...): Write a text summary.

trace_export(...): Stops and exports the active trace as a Summary and/or profile file.

trace_off(...): Stops the current trace and discards any collected information.

trace_on(...): Starts a trace to record computation graphs and profiling information.

write(...): Writes a generic summary to the default SummaryWriter if one exists.