我想在操作期间更改输出张量的基础存储。
我有新数据的原始指针(float *)。我想在启动内核并返回之前将输出张量设置为此新数据,以便可以劫持此操作。
但是我对何时删除原始指针感到困惑,因为张量构造似乎是一个浅表副本。我只能在所有此张量的使用完成后删除原始指针。但是如何通知我呢?
c++ tensorflow
c++ ×1
tensorflow ×1