Python 我可以修改Tensorflow自定义op中的输入吗？ 在TySoFrof中，当在C++中编写自定义OP时，一个提供了从上下文输入的指针。指针驻留在CPU或GPU上，具体取决于设备。我的问题是你能修改op中指针指向的内存缓冲区吗？例如，我能在GPU上进行就地

我可以修改Tensorflow自定义op中的输入吗？

---

在TySoFrof中，当在C++中编写自定义OP时，一个提供了从上下文输入的指针。指针驻留在CPU或GPU上，具体取决于设备。我的问题是你能修改op中指针指向的内存缓冲区吗？例如，我能在GPU上进行就地并行缩减吗？这会干扰其他操作的输入吗？

如果没有人回答，我建议您尝试一下。您最后是否尝试过此操作？取决于您正在做什么。如果你需要它作为后支撑，那么就不需要了。