Предположим, что у меня есть собственный 2D-слой в Keras, который принимает тензор формы (?, n_h, n_w) и создает тензор формы (?, m_h, m_w).
Как я могу пропустить каждый канал 3D-тензора формы (?, n_h, n_w, n_c) через этот слой, чтобы получить 3D-тензор формы (?, m_h, m_w, n_c)?
Обратите внимание, что пользовательский слой будет называться n_c time с другим набором весов.
Также обратите внимание, что я не хочу использовать цикл for для циклического прохождения каждого канала по одному.