Python Model Summary Tensorflow Tutorial Mnist