Could I implement my own convolution function to replace tf.nn.conv2d in tensorflow? How do I do?

Question

I want to implement my own convolution function just like tf.nn.conv2d function in tensorflow. I want to know whether I could do this with existed interface provided by tensorflow. Someone advised me to implement this according to the "how to add a new op in tensorflow" tutorial at the tensorflow website. Should I really implement my function in C++ firstly, then add it to tensorflow?

Answer 1

You might want to take a look at How to make a custom activation function with only Python in Tensorflow?

patapouf_ai explains how to add an operation using python code (without the need to implement C++ code.