1.导入所需模块:
import tensorflow as tf
import numpy as np
2.导入mnist数据集:
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
3.定义一个占位符,用来表示输入任意数量的mnist图像
x = tf.placeholder("float", [None, 784])
4.定义权重值 和 偏置量
w = tf.Variable(tf.zeros([784,10]))
b = tf.Variable(tf.zeros([10]))
5.softmax模型代码
y = tf.nn.softmax(tf.matmul(x,w) + b)