Повторите керас (тензор потока) модель по дополнительному измерению

Question

Допустим, у меня есть модель, которая отображает тензор с формой [n, 10] на тензор с формой [n, 2], где n - размер пакета. Как я могу повторить модель, чтобы полученная модель принимала входной тензор с формой [n, k, 10] и выводила тензор с формой [n, k, 2]? Версии k модели должны разделять все веса.

Answer 1

Вы можете сделать что-то вроде этого:

input_ = Input((k, model.input.shape[1]))
input_as_list = Lambda(lambda x: tf.unstack(x, axis=1))(input_)
model_outputs = [model(x) for x in input_as_list] 
model_outputs = [Lambda(lambda x: K.expand_dims(x, axis=1))(y) for y in model_outputs]
concat_output = Concatenate(axis=1)(model_outputs)
new_model = Model(input_, concat_output)