title: Problem showing the predicted image in Tensorflow mnist tutorial
body: <p>I'm working with the code from the Tensorflow <a href="https://www.tensorflow.org/tutorials" rel="nofollow noreferrer">beginner mnist tutorial</a> (code below).</p>
  
  <pre><code> import tensorflow as tf
   mnist = tf.keras.datasets.mnist
  
   (x_train, y_train),(x_test, y_test) = mnist.load_data()
   x_train, x_test = x_train / 255.0, x_test / 255.0
  
   model = tf.keras.models.Sequential([
     tf.keras.layers.Flatten(input_shape=(28, 28)),
     tf.keras.layers.Dense(512, activation=tf.nn.relu),
     tf.keras.layers.Dropout(0.2),
     tf.keras.layers.Dense(10, activation=tf.nn.softmax)
   ])
   model.compile(optimizer='adam',
                 loss='sparse_categorical_crossentropy',
                 metrics=['accuracy'])
  
   model.fit(x_train, y_train, epochs=5)
   model.evaluate(x_test, y_test)
  </code></pre>
  
  <p>Since the tutorial stops after evaluating the model, I have added a few lines to make actual predictions.</p>
  
  <pre><code> predictions = model.predict([x_test])
   print(np.argmax(predictions[0]))
  </code></pre>
  
  <p>So far it all works. However I would like to see the image the model tries to predict to know what's going on. I therefore tried the following:</p>
  
  <pre><code> import matplotlib.pyplot as plt
   plt.imshow(x_test[0])
   plt.show()
  </code></pre>
  
  <p>This does not work. I don't get an error message, it just won't show the picture. Furthermore if I try to add a simple print-statement after the matplotlib code that is ignored too and won't be printed. Any idea what's going on?</p>