Python Model Summary Tensorflow Tutorial Mnist Digit