from tensorflow.examples.tutorials.mnist import input_data

mnist = input_data.read_data_sets('MNIST_data', one_hot=True)
x_train = mnist.train.images
print ('x_train Examples Loaded = ' + str(x_train.shape))
y_train = mnist.train.labels
print ('y_train Examples Loaded = ' + str(y_train.shape))
x_test = mnist.test.images
print ('x_test Examples Loaded = ' + str(x_test.shape))
y_test = mnist.test.labels
print ('y_test Examples Loaded = ' + str(y_test.shape))