Керас выбрасывает ошибку формы, хотя я даю ей правильную форму

Question

У меня простая условная модель VAE.Во время выполнения я передаю шум декодера и метки

noise = np.random.normal(0, 1.0, size=(2,))  
decoder.predict([noise,labels])

, а затем получаю следующую ошибку:

ValueError: Error when checking : expected decoder_noise to have shape (2,) but got array with shape (1,)

На графике я также вижу, что он ожидаетмассив (2,), как показано на рисунке ниже.

Я что-то упускаю здесь из виду?

Примечание: labels имеет форму (4,), поэтому я не случайно меняю их.

Answer 1

Вам нужно задать двумерный массив шума, а не только (2,) для каждой имеющейся у вас метки.В этом случае ваш шум должен иметь значение size=(4,2), поэтому для каждого сэмпла вы получаете (2,), чего и ожидает модель.

Выводное сообщение состоит в том, что первое измерение сообщает количество сэмплов, а метки - скалярныетак что у вас есть (4,), но шума нет.Так что для каждого сэмпла вам нужен шум, следовательно (4,2).