mnist手写数据集（tensorflow）

全连接网络实现mnist

import tensorflow as tf
from tensorflow import keras
#  导入经典数据集加载模块
from tensorflow.keras import datasets
import matplotlib
from matplotlib import pyplot as plt


def preprocess(x, y):
    # 调用此函数时会自动传入 x,y 对象，shape 为[b, 28, 28], [b]
    # 标准化到 0~1
    x = tf.cast(x, dtype=tf.float32) / 255.
    x = tf.reshape(x, [-1, 28 * 28])  # 打平
    y = tf.cast(y, dtype=tf.int32)  # 转成整型张量
    y = tf.one_hot(y, depth=10)  # one-hot 编码
    # 返回的 x,y 将替换传入的 x,y 参数，从而实现数据的预处理功能
    return x, y;


#  数据加载
# 加载 MNIST 数据集
(x, y), (x_test, y_test) = datasets.mnist.load_data()
print('x:', x.shape, 'y:', y.shape, 'x test:', x_test.shape, 'y test:', y_test)

#  数据图片 x ,标签 y
#  构建 Dataset 对象
batchsz = 128
train_db = tf.data.Dataset.from_tensor_slices((x, y))
#  随机打散样本，不会打乱样本与标签映射关系
train_db