dataset = tf.keras.datasets.mnist
(X_train,y_train), (X_test,y_test) = dataset.load_data()
print(X_train.shape, y_train.shape)
print(X_test.shape, y_test.shape)

        __
        __