标签: python c++ tensorflow
在Tensorflow中,当人们用C ++编写自定义操作时,会为它提供一个指向上下文输入的指针。指针驻留在CPU或GPU上,具体取决于设备。我的问题是您可以修改op指针所指向的内存缓冲区吗?例如,我可以在GPU上进行并行还原吗?这会使其他操作的输入混乱吗?