Можно ли использовать выходы разной формы многоголовой модели Keras в одной функции потерь? - PullRequest
1 голос
/ 01 июня 2019

Мне интересно использовать несколько выходов из модели Keras в одной целевой функции.

Я не могу просто использовать объединенный слой, так как выходы имеют разный размер.В качестве примера: рассмотрим модель, которая будет выполнять стандартную задачу распознавания цифр в наборе данных MNIST.Мне нужна модель для вывода одного тензора размера (784,) (входная форма после сглаживания изображений) и одного размера (10,) (вероятности класса) и использования их обоих в одной пользовательской функции потерь.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...