环境:
1.Windows 10
2.CUDA9.2 + cuDNN v7.1.4 + Python 3.6.5 + TensorFlow 1.10.0 + TensorBoard 1.10.0
tensorflow_summary.py
import tensorflow as tf
x = tf.placeholder(tf.float32,shape=[])
y = tf.Variable(2,dtype = tf.float32)
assign_opt = tf.assign(y,y+1)
with tf.control_dependencies([assign_opt]):
out = x * y
init_opt = tf.global_variables_initializer()
with tf.Session() as sess:
sess.run(init_opt)
for _ in range(3):
res = sess.run(out,feed_dict = {x:1})
print("{} = {}".format(_,res))
with tf.summary.FileWriter(
logdir = r"G:\\AI\\tf_data\\tf_flush",\
graph = tf.get_default_graph()
) as writer:
writer.flush()
运行py代码:
执行可视化:
浏览器打开此地址效果如下: