2 выход нейронной сети как вход третьей нейронной сети

Question

Привет, я пытаюсь создать следующую схему нейронной сети, используя pytorch или keras, но я не знаю, как это сделать, может ли кто-нибудь помочь.

схема:

Схема

Answer 1

Здесь идет реализация Keras с использованием функционального API

from keras.models import Model
from keras.layers import Dense, Input, concatenate

def createModel( inp_1_shape, inp_2_shape):
  first_input = Input(shape = (inp_1_shape,))
  first_dense = Dense(1, )(first_input)

  second_input = Input(shape = (inp_2_shape,))
  second_dense = Dense(1, )(second_input)

  merge = concatenate([first_dense, second_dense])
  merge = Dense(2, )(merge)
  merge = Dense(3, )(merge)
  merge = Dense(1, )(merge)

  model = Model(inputs=[first_input, second_input], outputs=merge)
  model.compile(optimizer='adam', loss='mse', metrics=['accuracy'])

  return model

Просто вызовите эту функцию, и она вернет модель keras, вам может потребоваться дважды проверить количество нейронов в каждом слое, но кромечто все будет в порядке.

Счастливое обучение