Python Model Summary Tensorflow Tutorial Mnist Data