在张量流中同时训练两个网络

Question

想象x_1和x_2是神经网络的两个输入，而y_1和y_2是输出。然后应将y_1和y_2连接起来并传递到第二个网络中。第二个网络的输出是单个二进制标签z={0,1}。我有矩阵X=[x_1;x_2;...;x_2n]（第一个网络的输入）和标签Z = [z_1;z_2;...;z_n]（第二个网络的输出）。

如何在tensorflow中设置此结构？

Answer 1

假设您有两个分别称为model1和model2的模型，而您的输入x_1和x_2是model1的张量输入：

x = [x_1, x_2]
y = model1(x)
y = tf.keras.layers.Concatenate()(y)
z = model2(y)
combined_model = tf.keras.Model(x, z)