https://www.omegaxyz.com/2018/04/26/tensorflow_mnist_simple/