Tensorflow `Tensor OpKernelContext:：可变输入（整数索引，布尔锁保持）`

我一直在尝试编写一个自定义运算，并希望修改输入张量。我浏览了代码，在

op_kernel.cc

但是，返回类型是值，而不是引用。我尝试过对其进行修改，但效果并未反映在输入中

如果打算按值返回，那么作者为什么要使用

record\u tensor\u reference（tensor）

增加引用计数

---

（顺便说一句，我已经对代码进行了重新分解，因此我不一定需要更改op中的输入。但我仍然期待着答案）

如果输入数据在CPU上，并且操作员在GPU上注册，tf会将输入复制到GPU内存中，因此可变_输入的返回指针指向GPU的部分。因此，这种效果无法在输入中反映出来