您可以使用以下代码导入MNIST数据集：
from tensorflow.keras.datasets import mnist
# 加载MNIST数据集
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 输出训练集和测试集的形状
print('训练集：', x_train.shape, y_train.shape)
print('测试集：', x_test.shape, y_test.shape)
这段代码使用了TensorFlow的keras.datasets模块中的mnist函数来加载MNIST数据集。加载后，数据集被拆分为训练集和测试集，分别存储在(x_train, y_train)和(x_test, y_test)变量中。此外，代码还输出了训练集和测试集的形状。