原代码
更换了tensorflow1.0版本之后代码跟着up主的教程敲就可以了,这里面没什么需要改动的
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(inputs, in_size, out_size, activation_function=None):
# add one more layer and return the output of this layer
Weights = tf.Variable(tf.random.normal([in_size, out_size]))
biases = tf.Variable(tf.zeros([1, out_size]) + 0.1)
Wx_plus_b = tf.matmul(inputs, Weights) + biases
if activation_function is