Python Model Summary Tensorflow Tutorial Mnist Neural Network