Как написать собственный модуль слоя пула в тензорном потоке?

Я думал, что добавление нового модуля, который будет делать пул центра.

Я искал в коде тензорного потока, и есть файл с именемgen_nn_ops.py который внутренне вызывает функцию из другого файла, передавая параметры «Maxpool», «AvgPool» и т. д. для выполнения необходимых вычислений.

Я хочу сделать пул центра, который выбирает центральный элемент в окне. У меня есть готовый код для версий matlab и c ++, но мне нужно знать, как добавить новый модуль в TensorFlow для таких вычислений. Также, где установить код обратного распространения.

Ответы на вопрос(1)

Ваш ответ на вопрос