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



这是一个关于如何在TensorFlow中添加新操作的教程。它讨论了如何在python中创建和加载共享对象。

但是,有一些位于core/kernels中的操作不会生成共享对象。相反,TF将在dist-packages/tensorflow/python/ops中生成类似gen_some_ops.py的文件。

如果我想使用第二种方法来添加一个新的操作(比如zero_out),那么我应该怎么做(除了教程)?

谢谢。

核心操作符在_pywrap_tensorflow.so。如果你打算为核心做出贡献,他们就会去那里。但是,通常情况下,我们首先将它们接受到contrib目录中,然后在稍后的阶段移到核心中。最好在开始与我们协调之前提交github问题。

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

最新更新