tensorboard即一种可视化工具,可以形象的观看网络结构,使用起来也非常方便,即将想要展示的内容放在with tf.name_scope(''):下面即可,“ ”中填入结构名称。
代码如下:
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
# 载入数据集
mnist = input_data.read_data_sets("MNIST_data", one_hot=True)
# 每个批次的大小
batch_size = 100
# 计算一共有多少个批次
n_batch = mnist.train.num_examples // batch_size
# 命名空间
with tf.name_scope('input'):
# 定义两个placeholder
x = tf.placeholder(tf.float32, [None, 784], name='x-input')
y = tf.placeholder(tf.float32, [None, 10], name='y-input')
with tf.name_scope('layer'):
# 创建一个简单的神经网络
with tf.name_scope('wights'):
W = tf.Variable(tf.zeros([784, 10]), name='W')
with tf.name_scope('biases'):
b = tf.Variable(tf.zeros([10]), name='b')
with tf.name_scope(