Выполнение операции по размеру пакета в Keras и объединение с матрицами

#python #tensorflow #keras

Вопрос:

Если у меня есть нижеприведенные матрицы в 1 пакете, где каждый элемент-1 пример: a = np.array([[11,12,13], [21,22,23]])

Я хочу рассчитать сумму по строкам последних 2 элементов, что приведет к: new_mat = np.array([34,36])

а затем я хочу связать это с каждым элементом моей исходной матрицы a , чтобы получить: new_a = np.array([[11,12,13,34,36], [21,22,23,34,36]])

а затем подключитесь new_a к модели MLP в Керасе.

Можно ли выполнить вышеуказанные операции в Керасе ( tf.keras.layers.Add() например)? Причина, по которой я думаю сделать это в Keras вместо того, чтобы рассматривать это как отдельную предварительную обработку ( numpy например, используя), заключается в том, что я надеюсь, что слой Keras, сохраненный в модели, выполнит ту же операцию во время вывода, поэтому мне не нужно выполнять эту операцию предварительной обработки вне Keras до вывода.

Спасибо.