多输出单损模型

Question

我有模型为tensorflow.keras.models.Model类的子类。 call方法返回[output_1，ouput_2]，其中output_1和output_2具有不同的形状。如何打包两个输出以用于相同损失函数？ （自定义损失上的y_pred是调用方法返回的列表）

Answer 1

您是否需要神经网络输出为2个单独的输出？

相反，您可以将它们组合为一个输出，然后在使用其余应用程序中的数据时将它们分开。要合并它们，请在最后一层之后使用tf.keras.layers.concatenate层，这会将您的2个输出合并为1。这样，只需将1个向量传递给损失函数。