Tensorflow中使用tensorflow_dataset包里的tfds.load('mnist')加载的mnist数据