how to load mnist dataset in tensorflow code example

Example: tensorflow mnist dataset import

from tensorflow.keras.datasets import mnist
(X_train, Y_train), (X_test, Y_test) = mnist.load_data()