c++ - 编写自定义用户操作时如何引用内置操作

Question

我正在 tensorflow 中编写一个自定义操作，我想引用一个已经存在的操作，可以是我自己的用户操作之一，也可以是内置操作之一。有没有什么方法可以做到这一点，而不仅仅是复制粘贴我所指的其他操作的代码？

Answer 1

答案是“视情况而定”。只有一些内置操作以可重用的方式考虑在内。例如，component-wise ops 被实现为 reusable functors ，因此您可以#include "tensorflow/core/kernels/cwise_ops.h" 并使用 UnaryFunctor 或 BinaryFunctor 类，如果您需要借用他们的一些功能。

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