Имеем установленные модули matplotlib и tensorflow, хотим посмотреть на картинки из датасета mnist. Код:

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
import matplotlib.pyplot as plt
for i in range(20):
    data = x_train[i]
    plt.imshow(data)
    plt.show()

В результате будут нарисованы в виде графиков matplotlib первые 20 цифр из тренировочного датасета