在编写自定义用户操作时如何引用内置操作

Question

我在tensorflow中编写了一个自定义操作，我想引用一个已存在的操作系统，我自己的用户操作系统或其中一个内置操作系统。有没有办法做到这一点，而不只是复制粘贴我所指的其他操作的代码？

Answer 1

答案是“这取决于”。只有一些内置操作系统以可重用的方式考虑在内。例如，组件方式操作实现为reusable functors，因此如果您需要借用某些类，则可能#include "tensorflow/core/kernels/cwise_ops.h"并使用UnaryFunctor或BinaryFunctor类他们的功能。

大多数其他操作都将其实现内联到相应Compute()的{​​{1}}方法中。对于您自己的自定义操作，您可以自由地在自由函数中定义实现，或者使用继承来定义操作之间共享的常用功能。