如何将Tensorflow OpKernelallocate_persistent（）缓冲区附加和跟踪到OpKernel输出缓冲区

Question

我已经定义了新的OpKernel，并为一个缓冲区使用了allocate_persistent，而我想将其附加到OpKernel的输出缓冲区中。

//全局变量 PersistentTensorpersistentTensor 张量* objTensor

Compute（OpKernelContext * context）覆盖{ .... .. OP_REQUIRES_OK（上下文，上下文-> allocate_persistent（DT_FLOAT，形状， ＆persistentTensor，＆objTensor））; ... ... }

我能够在多个调用中保持persistentTensor，objTensor缓冲区并执行操作。但是无法将其附加到此op的输出（与我们在调用下面的allocate_output（）相同） OP_REQUIRES_OK（context，context-> allocate_output（0，out_shape，＆output1））;

由于这个原因，我无法将此OpKernel Op传递给下一个Op。

我有两个要求：

1. 如何将通过allocate_persistent（）调用创建的张量缓冲区附加到OpKernel的输出。这样我就可以在下一次调用OpKernel中将其用作输入
2. 在下一次用作输入缓冲区时，如何获取有关该持久缓冲区的信息。基于此，我将其用于其他目的。

预先感谢