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

2024-05-28 19:40:50 发布

您现在位置:Python中文网/ 问答频道 /正文

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


Tags: 内存gpucpu缓冲区指向指针optysofrof

热门问题