在 TensorFlow 中，您可以出于各种原因创建自定义分配器(我正在为新硬件执行此操作)。由于设备的结构，我需要使用一些元素的结构作为我的数据指针，分配器将其作为 void* 返回。
在我正在编写的内核中，我可以访问张量，但我需要获取我编写的指针结构。检查这些类，似乎我可以通过执行 tensor_t.buf_->data()
问题是我找不到执行此操作的代码，并且我担心它不安全(很有可能!)或者有更标准的方法来执行此操作。
有人可以确认这是否是一个好/坏主意吗？并提供替代方案(如果存在)？
最佳答案
您或许还可以使用Tensor::tensor_data() .data() 来访问原始指针，而无需通过 DMAHelper 使用奇怪的间接寻址。
关于c++ - TensorFlow 自定义分配器和从 Tensor 访问数据，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/39797095/