Как смоделировать общий слой в керасе?

Question

Я хочу обучить модель с общим слоем в следующем виде:

x --> F(x)
          ==> G(F(x),F(y))
y --> F(y) 

x и y - два отдельных входных слоя, а F - общий слой.G - последний слой после объединения F(x) и F(y).

Возможно ли смоделировать это в Керасе?Как?

Answer 1

Вы можете использовать Функциональный API Keras для этой цели:

from keras.layers import Input, concatenate

x = Input(shape=...)
y = Input(shape=...)

shared_layer = MySharedLayer(...)
out_x = shared_layer(x)
out_y = shared_layer(y)

concat = concatenate([out_x, out_y])

# pass concat to other layers ...

Обратите внимание, что x и y могут быть выходными тензорами любого слоя и необязательно входными слоями.