from __future__ import absolute_import, division, print_function
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers, optimizers, datasets
import matplotlib.pyplot as plt
(train_images, train_labels), (test_images, test_labels) = keras.datasets.mnist.load_data()
# 载入mnist数据集
train_images = train_images.reshape((-1,28,28,1))
test_images = test_images.reshape((-1,28,28,1))
# 对数据格式进行改造
model = keras.Sequential([
# layers.Conv2D(input_shape=(train_images.shape[1], train_images.shape[2], train_images.shape[3]), filters=32, kernel_size=(3,3), strides=(1,1), padding='valid', activation='relu'),
layers.Conv2D(input_shape=(train_images.shape[1], train_images.shape[2], train_images.shape[3]),filters=32, kernel_size=[3, 3], padding="valid",