Определение моделей Keras для массивов

Question

У меня есть 3 входа, которые войдут в мою модель. Это: 1. 20 X 20 Список 2. n чисел массива X [2] или как [[a, b], [c, d], ......] 3. [a, b] array

Вот и все, что делает модель простой, но я не понимаю, как определить входы и выходы для этого.

Модель должна иметь 2 скрытых слоя по 32 в каждом.

    inputBoard = tf.keras.Input(shape=(20,20))
    x1 = tf.keras.layers.Flatten()(inputBoard)
    boardDense = tf.keras.layers.Dense(32, activation='relu')(x1)

    inputAgent = tf.keras.Input(shape=(None, 2))
    x2 = tf.keras.layers.Flatten()(inputAgent)
    agentDense = tf.keras.layers.Dense(16, activation='relu')(x2)

    inputReward = tf.keras.Input(shape=(1,1))
    x3 = tf.keras.layers.Flatten()(inputReward)
    rewardDense = tf.keras.layers.Dense(4, activation='relu')(x3)

    concat = tf.keras.layers.concatenate([boardDense, agentDense, rewardDense])

    hidden = tf.keras.layers.Dense(32, activation = 'relu')(concat)
    hidden2 = tf.keras.layers.Dense(32, activation='relu')(hidden)

    output = tf.keras.layers.Dense(4, activation="softmax")(hidden2)

    self.model = tf.keras.Model(inputs=[inputBoard, inputAgent, inputReward], outputs = output)

    self.model.build()
    optimizer = tf.keras.optimizers.Adam(lr = self.learningRate)
    self.model.compile(loss = 'mse', optimizer = optimizer)
    self.model.summary()