Sáp nhập hai mô hình khác nhau trong Keras

Tôi đang cố gắng hợp nhất hai mô hình Keras thành một mô hình duy nhất và tôi không thể thực hiện được điều này.

Ví dụ, trong Hình đính kèm, tôi muốn tìm nạp lớp giữa của chiều 8 và sử dụng lớp này làm đầu vào cho lớp (của chiều thứ 8 một lần nữa) trong Mô hình và sau đó kết hợp cả Mô hình và Mô hình làm một mô hình.$A2$$B1$$B$$A$$B$

Tôi đang sử dụng mô-đun chức năng để tạo Model và Model một cách độc lập. Làm thế nào tôi có thể hoàn thành nhiệm vụ này?$A$$B$

Lưu ý : là lớp đầu vào cho mô hình và là lớp đầu vào cho mô hình .$A1$$A$$B1$$B$

Tôi đã tìm ra câu trả lời cho câu hỏi của tôi và đây là đoạn mã được xây dựng dựa trên câu trả lời trên.

from keras.layers import Input, Dense
from keras.models import Model
from keras.utils import plot_model

A1 = Input(shape=(30,),name='A1')
A2 = Dense(8, activation='relu',name='A2')(A1)
A3 = Dense(30, activation='relu',name='A3')(A2)

B2 = Dense(40, activation='relu',name='B2')(A2)
B3 = Dense(30, activation='relu',name='B3')(B2)

merged = Model(inputs=[A1],outputs=[A3,B3])
plot_model(merged,to_file='demo.png',show_shapes=True)

và đây là cấu trúc đầu ra mà tôi muốn:

Trong Keras có một cách hữu ích để xác định mô hình: sử dụng API chức năng . Với API chức năng, bạn có thể xác định một đồ thị theo chu kỳ theo hướng, cho phép bạn xây dựng các kiến ​​trúc hoàn toàn tùy ý. Xem xét ví dụ của bạn:

#A_data = np.zeros((1,30))
#A_labels = np.zeros((1,30))
#B_labels =np.zeros((1,30))

A1 = layers.Input(shape=(30,), name='A_input')
A2 = layers.Dense(8, activation='???')(A1)
A3 = layers.Dense(30, activation='???', name='A_output')(A2)


B2 = layers.Dense(40, activation='???')(A2)
B3 = layers.Dense(30, activation='???', name='B_output')(B2)

## define A
A = models.Model(inputs=A1, outputs=A3)

## define B
B = models.Model(inputs=A1, outputs=B3) 

B.compile(optimizer='??',
          loss={'B_output': '??'}
          )

B.fit({'A_input': A_data},
  {'B_output': B_labels},
  epochs=??, batch_size=??)

À chính nó đấy! Bạn có thể xem kết quả bằng cách B.summary()::

Layer (type)                 Output Shape              Param    
A_input (InputLayer)         (None, 30)                0         
_________________________________________________________________
dense_8 (Dense)              (None, 8)                 248     
______________________________________________________________
dense_9 (Dense)              (None, 40)                360       
_________________________________________________________________
B_output (Dense)             (None, 30)                1230