tensorflow可视化

看一下加的主要操作，如下：

tf.summary.scalar

tf.summary.merge_all()

tf.summary.FileWriter(’./result’, sess.graph)

writer.add_summary(summary, i)

# tensorflow可视化
with tf.device("/cpu:0"):
    with tf.variable_scope("foo"):
        x_init1 = tf.get_variable('init_x', [10], tf.float32, initializer=tf.random_normal_initializer())[0]
        x = tf.Variable(initial_value=x_init1, name='x')
        y = tf.placeholder(dtype=tf.float32, name='y')
        z = x + y

    with tf.variable_scope("bar"):
        a = tf.constant(3.0) + 4.0

    w = z * a

# 开始记录信息(需要展示的信息的输出)
tf.summary.scalar(name='scalar_init_x', tensor=x_init1)
tf.summary.scalar(name='scalar_x', tensor=x)
tf.summary.scalar(name='scalar_y', tensor=y)
tf.summary.scalar(name='scalar_z', tensor=z)
tf.summary.scalar('scalar_w', w)  # name=和tensor=可以不写，但必须同时不写
# update x
assign_op = tf.assign(x, x + 1)
with tf.control_dependencies([assign_op]):
    with tf.device('/gpu:0'):
        out = x * y
    tf.summary.scalar(name='scalar_out', tensor=out)

with tf.Session(config=tf.ConfigProto(log_device_placement=True, allow_soft_placement=True)) as sess:
    # 下面两行代码可以写到Session外面
    # merge all summary
    merged_summary = tf.summary.merge_all()  # 触发所有的输出操作
    # 得到输出到文件的对象
    writer = tf.summary.FileWriter('./result', sess.graph)

    # 初始化
    sess.run((tf.global_variables_initializer()))
    print(sess.run(x))  # -1.7564569
    # print
    for i in range(1, 5):
        summary, r_out, r_x, r_w = sess.run([merged_summary, out, x, w], feed_dict={y: i})
        writer.add_summary(summary, i)  # 把每一次run的信息和得到的数据加到writer里面
        print("{},{},{}".format(r_out, r_x, r_w))
        ''
        -0.7564568519592285,-0.7564568519592285,-5.2951979637146
        0.48708629608154297,0.24354314804077148,8.704801559448242
        3.7306294441223145,1.2435431480407715,22.704801559448242
        8.974172592163086,2.2435431480407715,36.704803466796875
        ''
        # 关闭操作
        writer.close()