Средний слой, который поставляется с Keras, уже имеет поддержку маскировки, однако, глядя на исходный код Average Layer, мне не ясно, как и если применяется маскирование.
У меня естьсписок входов, каждый со своей маскировкой (например, из слоя внедрения). Средний слой, который я хочу, должен взять среднее из тех входных данных, которые не были замаскированы. Другими словами, если вход маскируется, он не должен иметь никакого значения в вычисленном среднем значении. Если все входы замаскированы, то выход замаскирован и передан следующим слоям.
Смежный вопрос заключается в том, что Средний уровень, который поставляется с библиотекой, поддерживает только функции слияния списка входных данных. Есть ли поддержка библиотеки для слияния тензора по определенному измерению? Можно ли нарезать тензор в списке входов для подачи в средний слой? Если нет, то как взять среднее значение тензора по некоторому измерению при наличии маскирования?
Я склоняюсь к написанию пользовательского среднего слоя, который вычисляет маски и использует маски при расчете выходных данных, но из документации, не понятно, как это сделать?
Любые указатели или примеры кода высоко ценятся.