Пользовательский оконный оператор в tenorflow

Question

Я хочу реализовать следующую формулу: Si = (x1 - w1) + ... + (xn - wn) как Conv = x1 * w1 + ... xn * wn для некоторой области под X (входной тензор) и ядро ​​W. Важно, чтобы эта операция повторялась на всех участках X со срезом W с параметрами шага и отступа, как простая свертка.

Как это сделать?

Некоторое время назад я нашел похожий вопрос по stackoverflow. Но он закончился на пользовательской реализации на C ++ и компиляции или изменении исходного кода CUDA или чего-то в этом роде.

Есть ли сегодня более легкий путь?

Answer 1

Если я правильно вас понял, тогда вы вычислите Si=(x1 + ... + xn)-(w1 + ... + wn)? Сумма весов - это одно число, поэтому у вас больше нет «ядра». Первую сумму вы можете вычислить с помощью tf.nn.conv2d и фильтра, который инициализируется с помощью tf.ones. Но я не думаю, что это то, что вы хотели сделать, поэтому не могли бы вы уточнить свой вопрос?