Tensorflow cnn MNIST 笔记
写这个完全是记录看官网example时不懂,但后来弄懂的一些细节。当然这个可以算是对官方文档的补充,也许每个人遇到的不懂都不一样,但希望对大家有帮助。
先上代码
#!/usr/bin/env python
from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
import numpy as np
def weightVariable(shape):
initial = tf.truncated_normal(shape,stddev=0.1)
return tf.Variable(initial)
def biasVariable(shape):
initial = tf.constant(0.1,shape=shape)
return tf.Variable(initial)
def conv2d(x,W):
return tf.nn.conv2d(x,W,strides=[