train和test同时纯在,test用train的变量,train=False,reuse=True,summary变量时只调用trainModel的summary函数,Writer用两个分别写
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
def add_layer(input, in_size, out_size, name='layer', activation=None):
with tf.variable_scope(name):
Weight = tf.get_variable(name='Weight', shape=(in_size, out_size), dtype=tf.float32, initializer=tf.random_normal_initializer())
bias = tf.get_variable(name='bias', shape=(out_size), dtype=tf.float32, initializer=tf.constant_initializer(0.1))
Wx_b = tf.matmul(input, Weight) + bias
if</