Модель множественных выходных одиночных потерь

Question

У меня есть модель, являющаяся подклассом класса tensorflow.keras.models.Model. Метод call возвращает [output_1, ouput_2], где output_1 и output_2 имеют разные формы. Как я могу упаковать оба выхода для использования в функции потерь той же ? (Пусть y_pred на настраиваемой потере будет списком, возвращаемым методом вызова)

Answer 1

Вам обязательно нужно, чтобы нейронный выход net был 2 отдельными выходами?

Вместо этого вы можете объединить их в один вывод, а затем разделить их позже, когда вы будете использовать данные в остальной части вашего приложения. Чтобы объединить их, используйте слой tf.keras.layers.concatenate после ваших последних слоев, который объединит ваши 2 выхода в 1. Таким образом, единственный вектор должен быть передан в функцию потерь.