注意：如果提供的代码是独立运行的，那么必须确保train_images和test_images是numpy数组（或类似数组，支持除以浮点数）。原始MNIST数据集的形状可能是(60000, 28, 28)和(10000, 28, 28)，分别代表训练集和测试集的图像数量、图像高度和图像宽度。因此，这里将训练集和测试集的图像数据重塑为(60000, 28, 28, 1)和(10000, 28, 28