我可以在Tensorflow自定义操作中修改输入吗?

时间:2018-09-12 00:56:56

标签: python c++ tensorflow

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

0 个答案:

没有答案