ошибка: только целочисленные скалярные массивы могут быть преобразованы в скалярный индекс с помощью numpy .reshape для набора данных tenorflow

Question

Я пишу программу, которая анализирует набор данных Cifar-100 с использованием TensorFlow и набора данных Keras. Я следую этому уроку

https://www.tensorflow.org/tutorials/keras/classification

Но я адаптировал его для cifar-100, а не для Fashion MNIST. Fashion MNIST - это набор данных с размером (60000, 28, 28), в то время как Cifar-100 имеет размер (50000, 32, 32, 3).

Поскольку мне нужно, чтобы мой train_images был в трех измерениях, я попытался написать

train_images.transpose(50000, 32, 32, 3) train_images.reshape(train_images, (50000, 32, 32))

, но есть AxisError: axis 50000 is out of bounds for array of dimension 4

поэтому я попытался использовать только функцию изменения формы: train_images.reshape(train_images, (50000, 32, 32)) но я получаю это TypeError: only integer scalar arrays can be converted to a scalar index.

Спасибо за помощь:)