Вы можете изменить свой ввод следующим образом (заменить num_of_rows
):
input = X.reshape((num_of_rows, 32 * 32))
А затем с помощью input_shape
указать форму керасу:
model.add(Dense(256, input_shape=(32 * 32,)))
Вы даже можете перейти и преобразовать значения в шкалу 0-1:
input = input.astype('float32') / 255