Keras - Репликация 1D тензора для создания 3D тензора

Question

В настоящее время я пытаюсь разработать модель генератора, которая будет принимать входное изображение и условную метку, объединенную в изображение.

Входное изображение в 3D (включая канал), а метка в 1D (содержит idx класса)

input_image = Input(shape=(row, col, chann))
cond = Input(shape=(1,))

На основании информации Объединение мне нужно изменить форму cond, чтобы она была в той же форме, что и input_image. Но, как я обнаружил в документах Keras, есть только RepeatVector, который повторяет 2D-тензор в 3D.

Как мне воспроизвести входной сигнал cond, чтобы он имел ту же форму, чтобы я мог объединить его?

Answer 1

Вы можете объединить Reshape и RepeatVector

input_image = Input(shape=(row, col, chann))
cond = Input(shape=(1,))
cond = Reshape((row,col,1))(RepeatVector(row*col)(cond))
concat = Concatenate()([input_image, cond])