# show the first bunch of training data
for i in range(10):
print(f'train image #{i}: label {train_labels[i]}')
- plt.imshow(train_images[i].reshape(rows, cols), cmap='gray')
+ plt.imshow(train_images[:, i].reshape(rows, cols), cmap='gray')
plt.show()