编写自定义用户操作时如何引用内置操作
How to refer to a built-in op when writing a custom user op
我正在 tensorflow 中编写一个自定义操作，我想引用一个已经存在的操作，可以是我自己的用户操作之一，也可以是内置操作之一。有没有什么方法可以做到这一点，而不仅仅是复制粘贴我所指的其他操作的代码？
答案是 "it depends." 只有一些 built-in 操作以可重用的方式考虑在内。例如，component-wise 操作被实现为 reusable functors，因此您可以 #include "tensorflow/core/kernels/cwise_ops.h" 并使用 UnaryFunctor 或 BinaryFunctor 类 如果你需要借用他们的一些功能。
大多数其他操作都将其实现内联到相应 OpKernel 的 Compute() 方法中。对于您自己的自定义操作，您可以自由地在自由函数中定义实现，或者使用继承来定义操作之间共享的通用功能。
我正在 tensorflow 中编写一个自定义操作，我想引用一个已经存在的操作，可以是我自己的用户操作之一，也可以是内置操作之一。有没有什么方法可以做到这一点，而不仅仅是复制粘贴我所指的其他操作的代码？
答案是 "it depends." 只有一些 built-in 操作以可重用的方式考虑在内。例如，component-wise 操作被实现为 reusable functors，因此您可以 #include "tensorflow/core/kernels/cwise_ops.h" 并使用 UnaryFunctor 或 BinaryFunctor 类 如果你需要借用他们的一些功能。
大多数其他操作都将其实现内联到相应 OpKernel 的 Compute() 方法中。对于您自己的自定义操作，您可以自由地在自由函数中定义实现，或者使用继承来定义操作之间共享的通用功能。