Как написать собственный модуль слоя пула в тензорном потоке?
Я думал, что добавление нового модуля, который будет делать пул центра.
Я искал в коде тензорного потока, и есть файл с именемgen_nn_ops.py
который внутренне вызывает функцию из другого файла, передавая параметры «Maxpool», «AvgPool» и т. д. для выполнения необходимых вычислений.
Я хочу сделать пул центра, который выбирает центральный элемент в окне. У меня есть готовый код для версий matlab и c ++, но мне нужно знать, как добавить новый модуль в TensorFlow для таких вычислений. Также, где установить код обратного распространения.