tensorflow 2.0版本如何import mnist数据集

最近在写代码时，需要调用MNIST数据集，好久没用tensorflow写代码了，才发现版本已经更新到2.0版本了。按照网上给的代码调用数据集，过程是如下所示：

from tensorflow.contrib.learn.python.learn.datasets.mnist import read_data_sets

# load mnist dataset
mnist = read_data_sets("./dataset", one_hot=True)

才发觉tensorflow 2.0版本中已经舍弃了contrib包了，搞得就很烦。找了很多资料，发现都不靠谱，好多博主都是直接叫你先下载本地MNIST数据集，这不就是真~人工智能吗？

后来终于找到一个理智的解决方法了，改进如下：

import tensorflow.keras as keras

# load mnist dataset
mnist = keras.datasets.mnist.load_data()