編寫自定義用戶操作時如何引用內置操作

Question

我正在用tensorflow編寫自定義操作，我想引用一個已經存在的操作，或者是我自己的用戶操作之一，或者是內置操作之一。 有什么方法可以做到，而不僅僅是復制我所指的另一個操作中的代碼？

Answer 1

答案是“視情況而定”。 只有某些內置操作以可重用的方式進行分解。 例如，基於組件的操作被實現為可重用的函子 ，因此，如果您需要借用某些功能，則可以#include "tensorflow/core/kernels/cwise_ops.h"並使用UnaryFunctor或BinaryFunctor類。

其他大多數操作都將其實現內聯到相應OpKernel的Compute()方法中。 對於自己的自定義操作，您可以自由地在自由函數中定義實現，也可以使用繼承定義操作之間共享的通用功能。