我可以实现自己的卷积函数来替换张量流中的tf.nn.conv2d吗？我该怎么办？

Question

我想实现我自己的卷积函数，就像tensorflow中的tf.nn.conv2d函数一样。我想知道我是否可以使用tensorflow提供的现有界面来做到这一点。有人建议我按照＆＃34;如何在张量流中添加新操作来实现这个＆＃34;在tensorflow网站上的教程。我是否应该首先在C ++中实现我的函数，然后将其添加到tensorflow？

Answer 1

您可能需要查看How to make a custom activation function with only Python in Tensorflow?

patapouf_ai解释了如何使用python代码添加操作（无需实现C ++代码。