TensorFlow实现基础CNN,两层卷积+2层全连接网络demo,代码如下,注释都有
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
=
50
#共有xx个批次
n_batch
=mnist.train.num_examples
//batch_size
#初始化权重
def
weight_variable(
shape):
#生成shape结构的变量,方差=0.1
initial
=tf.truncated_normal(shape,
stddev
=
0.1)
return tf.Variable(initial)
#初始化偏置
def
bias_variable(
shape):
initial
=tf.constant(
0.1,
shape
=shape)
return