import numpy as np
from matplotlib import pyplot as plt

with open('./mnist/train-images.idx3-ubyte', 'rb') as images_file,\
        open('./mnist/train-labels.idx1-ubyte', 'rb') as labels_file: