tf.scalar_summary(tags, values)
# ...
summary_op = tf.summary.merge_all() #向计算图中添加记录节点
summary_writer = tf.summary.FileWriter(logdir, graph=sess.graph)
summary_str = sess.run(summary_op) #运行记录节点
summary_writer.add_summary(summary_str, global_step) #将运行结果写入日志书写器实例对象中
简化后可以这样:
summary_writer = tf.summary.FileWriter(logdir)
#实例化一个tf.Summary()类
summary = tf.Summary(value=[tf.Summary.Value(tag="summary_tag", simple_value=0),
tf.Summary.Value(tag="summary_tag2", simple_value=1)])
summary_writer.add_summary(summary, x)
或
summary_writer = tf.summary.FileWriter(logdir)
summary = tf.Summary() #实例化一个tf.Summary()类
summary.value.add(tag="summary_tag", simple_value=0)
summary.value.add(tag="summary_tag2", simple_value=1)
summary_writer.add_summary(summary, x)
下面看一段完整代码:
import tensorflow as tf
summary_writer = tf.summary.FileWriter('/tmp/test')
summary = tf.Summary(value=[
tf.Summary.Value(tag="summary_tag", simple_value=0),
tf.Summary.Value(tag="summary_tag2", simple_value=1),
])
summary_writer.add_summary(summary, 1)
summary = tf.Summary(value=[
tf.Summary.Value(tag="summary_tag", simple_value=1),
tf.Summary.Value(tag="summary_tag2", simple_value=3),
])
summary_writer.add_summary(summary, 2)
summary_writer.close()