我有一个关于测试模型的问题。我使用tf.keras.utils.image_dataset_from_directory创建了一个模型测试集，如下所示：

batch_size = 32
test_dataset = tf.keras.utils.image_dataset_from_directory(
    '/content/drive/MyDrive/test',
    image_size = (224, 224),
    batch_size = batch_size,
    shuffle = False
)

然后我得到的输出是Found 150 files belonging to 3 classes.之后，我想使用以下语句迭代测试数据集批处理：

labels_batch = []
for dataset in test_dataset.unbatch():
  image_batch, label_batch = dataset
  labels = label_batch.numpy()
  labels_batch.append(labels)

我了解到，在数据集〈类'元组'〉的结构中，由两个位置组成，分别是image_batch和label_batch，它们是**〈类'tensorflow.python.framework.ops. EagerTensor'〉**。
因此，image_batch[0]应表示test_dataset中的第一个图像。当我想显示第一个图像的数组时，我使用命令print(image_batch[0])，如所示，所有图像的数组为shape=(224, 3)，但我认为所有图像的大小应为shape=(224,224,3)。那么，我必须使用什么命令来访问每个图像的数组？
我在google colab中使用TensorFlow版本2.9。我不确定test_dataset.unbatch()。问题是否在这里？