from tensorflow.examples.tutorials.mnist import input_data
import numpy as np
import matplotlib.pyplot as plt

a=batch_xs[20,:]

first_train_img = np.reshape(a, (28, 28))
plt.matshow(first_train_img, cmap = plt.get_cmap('Blues'))
plt.show()