这是一个关于如何在TensorFlow中添加新操作的教程。它讨论了如何在python中创建和加载共享对象。
但是,有一些位于core/kernels中的操作不会生成共享对象。相反,TF将在dist-packages/tensorflow/python/ops
中生成类似gen_some_ops.py
的文件。
如果我想使用第二种方法来添加一个新的操作(比如zero_out
),那么我应该怎么做(除了教程)?
谢谢。
核心操作符在_pywrap_tensorflow.so
。如果你打算为核心做出贡献,他们就会去那里。但是,通常情况下,我们首先将它们接受到contrib
目录中,然后在稍后的阶段移到核心中。最好在开始与我们协调之前提交github问题。
如果您不打算回馈,那么user_ops
路径是最合适的。这样你就可以有一定程度的版本独立性。