可以在单个损失函数中使用多头Keras模型的不同形状的输出吗？

Question

我有兴趣在单个目标函数中使用Keras模型的多个输出。

我不能简单地使用连接层，因为输出的大小不同。例如：考虑一个将在MNIST数据集上执行标准数字识别任务的模型。我需要模型输出一个张量大小（784，）（图像被展平后的输入形状）和一个张量大小（10，）（类概率），并在单个自定义损失函数中使用它们。