我想我可以使用 TF_NewTensor(..., void* data,...)
函数(来自 Tensorflow C API)通过 data
指向共享内存。如果我将该张量传递给 tf.Vaiable()
,Tensorflow 变量是否会将其数据存储在共享内存中并从中读取?如果没有,是否有可能获得指向变量底层数据缓冲区的指针?
最佳答案
目前,TensorFlow 需要拥有 Tensor 缓冲区,因为系统会根据它们的使用情况在内部重新计算它们。我们没有将现有缓冲区的所有权转移到 Tensor 的 API,但请随时在 GitHub 上为此提交功能请求。
关于可以从共享内存创建 Tensorflow 变量吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38317380/