1.3　JAX实战——MNIST手写体的识别

MNIST是深度学习领域常见的数据集。每一个MNIST数据单元由两部分组成：一幅包含手写数字的图片和一个对应的标签。我们把这些图片设为“xs”，把这些标签设为“ys”。训练数据集和测试数据集都包含xs和ys，比如训练数据集的图片是mnist_train_x，训练数据集的标签是mnist_train_y。

如图1.21所示，每一幅图片包含28×28个像素点。如果我们把这个数组展开成一个向量，长度是28×28 = 784。如何展开这个数组（数字间的顺序）不重要，只要保持各幅图片采用相同的方式展开即可。从这个角度来看，MNIST数据集的图片就是在784维向量空间里面的点。

图1.21　数据集MNIST