TensorBoard可视化

import tensorflow as tf


#参数概要
def variable_summaries(var):
    with tf.name_scope('summaries'):
        mean = tf.reduce_mean(var)
        tf.summary.scalar('mean', mean)#平均值
        with tf.name_scope('stddev'):
            stddev = tf.sqrt(tf.reduce_mean(tf.square(var - mean)))
        tf.summary.scalar('stddev', stddev)#标准差
        tf.summary.scalar('max', tf.reduce_max(var))#最大值
        tf.summary.scalar('min', tf.reduce_min(var))#最小值
        tf.summary.histogram('histogram', var)#直方图

W = tf.Variable(tf.zeros([784,10]),name='W')
variable_summaries(W) #函数可以记录权值的图，减少重复运算

with tf.name_scope('summaries'):
     mean = tf.reduce_mean(var)
     tf.summary.scalar('mean', mean)#平均值
    #tf.summary.histogram('histogram', var)#直方图

#合并所有的summary
merged = tf.summary.merge_all()   

with tf.Session() as sess:
    sess.run(init)
    writer = tf.summary.FileWriter('logs/',sess.graph)
    for epoch in range(51):
        for batch in range(n_batch):
            batch_xs,batch_ys =  mnist.train.next_batch(batch_size)
            summary,_ = sess.run([merged,train_step],feed_dict={x:batch_xs,y:batch_ys})

    writer.add_summary(summary,epoch)#每迭代一次，记录一次summary的值

进入logs所在盘：cmd中输入 ：tensorboard  --logdir=文件所在路径