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?

score 2 · Accepted Answer · edited May 23 '17 at 12:17

2

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.

edited May 23 '17 at 12:17

Community

1
1

answered Apr 30 '17 at 09:35

Roy Jevnisek

320
1
8

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

1 Answers1