Python Data Augmentation Tensorflow Tutorial Mnist