小编Cor*_*Rex的帖子

如何在C ++中从原始指针数据构造Tensorflow :: Tensor

我想在操作期间更改输出张量的基础存储。

我有新数据的原始指针(float *)。我想在启动内核并返回之前将输出张量设置为此新数据,以便可以劫持此操作。

但是我对何时删除原始指针感到困惑,因为张量构造似乎是一个浅表副本。我只能在所有此张量的使用完成后删除原始指针。但是如何通知我呢?

c++ tensorflow

5
推荐指数
1
解决办法
1967
查看次数

标签 统计

c++ ×1

tensorflow ×1