TensorFlow2.0如何导入数据集？

tensorflow 1.x 导入数据集方法：

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

但是在新版本（tensorflow 2.0这个方法就报错了）
参考文档：

【TensorFlow2.0】以后我们再也离不开Keras了？
正确导入方法：

from tensorflow import  keras
from tensorflow.keras.datasets import mnist

(X_tarin, y_train), (X_test, y_test) = mnist.load_data()

在TensorFlow2.0中，常用的数据集需要使用tf.keras.datasets来加载，在datasets中有如下数据集。

对于上图中的数据集我们可以像下面这样加载