Tensorflow 2.0 Mnist Example