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

Question

我正在張量流中編寫自定義操作，我想引用已存在的操作，或者是我自己的用戶操作，或者其中一個內置操作。有沒有辦法做到這一點，而不僅僅是複製粘貼來自我指的另一個操作的代碼？

Answer 1

對此的答案是「它取決於」。只有一些內置的操作系統以可重用的方式進行分解。例如，基於組件的操作實現爲reusable functors，所以如果您需要借用其某些功能，您可能可以使用#include "tensorflow/core/kernels/cwise_ops.h"並使用UnaryFunctor或BinaryFunctor類。

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