如何将CIFAR10图像可视化为矩阵

import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf

(x_train, y_train), (x_test, y_test) = tf.keras.datasets.cifar10.load_data()

x_train = x_train / 255.0
x_test = x_test / 255.0

class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer',
               'dog', 'frog', 'horse', 'ship', 'truck']

y_train = np.array([class_names[y[0]] for y in y_train])
y_test = np.array([class_names[y[0]] for y in y_test])

image_index = 0
image_matrix = x_train[image_index]

plt.imshow(image_matrix)
plt.title(y_train[image_index])
plt.axis('off')
plt.show()