我想我可以使用该TF_NewTensor(..., void* data,...)
函数(来自Tensorflow C API)通过data
指向共享内存来从共享内存创建新的张量。如果我将该张量传递给tf.Vaiable()
,Tensorflow变量是否会将其数据存储在共享内存中并从共享内存中读取数据?如果不是,是否有可能获得指向变量基础数据缓冲区的指针?
目前,TensorFlow需要拥有Tensor缓冲区,因为系统会根据使用情况在内部对其进行重新引用。我们没有API将现有缓冲区的所有权转让给Tensor,尽管您可以随时在GitHub上提交功能请求。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句