使用与教程不同的方法在Tensorflow中添加新操作

Question

以下是关于how to add a new operation in TensorFlow的教程。它讨论了如何创建一个共享对象并在python中加载它。

但是，有些操作位于核心/内核中，没有生成共享对象。相反，TF将在dist-packages/tensorflow/python/ops中生成像gen_some_ops.py这样的文件。

如果我想用第二种方法来添加一个新的操作（比如zero_out），那么我应该怎么做（除了教程）？

谢谢。

Answer 1

核心运营商在_pywrap_tensorflow.so。如果你打算为核心作出贡献，那就是他们将要去的地方。不过，通常情况下，我们会先将它们接受到contrib目录中，然后在稍后阶段进入核心。在开始与我们协调之前，最好先提交github issue。

如果您不打算回馈，那么user_ops路径是最合适的。这样你可以有一定程度的版本独立性。