Tensorflow 在计算方法中访问的张量数据(如flat()或vec())的生存期是多少?
如果我在OpKernel的Compute()方法中并访问输入张量,例如:Tensorflow 在计算方法中访问的张量数据(如flat()或vec())的生存期是多少?,tensorflow,Tensorflow,如果我在OpKernel的Compute()方法中并访问输入张量,例如: auto data = context->input(0).vec<string>(); autodata=context->input(0.vec(); 数据的生命周期是多少?我是否可以将该对象存储在类状态变量中,并期望数据在后续运行中保持有效 如果不是,复制数据的正确内存管理技术是什么 最好使用指向示例的指针。数据对象是一个Eigen::TensorMap,其中包含指向对象中原始缓冲区的借用指针。
auto data = context->input(0).vec<string>();
autodata=context->input(0.vec();
数据的生命周期是多少?我是否可以将该对象存储在类状态变量中,并期望数据在后续运行中保持有效
如果不是,复制数据的正确内存管理技术是什么
最好使用指向示例的指针。数据
对象是一个Eigen::TensorMap
,其中包含指向对象中原始缓冲区的借用指针。必须确保未销毁tensorflow::Tensor
对象。最简单的方法是将tensorflow::Tensor
存储为类中的成员变量。(tensorflow::Tensor本身包含一个引用计数的缓冲区,因此“复制”该类型的对象相对便宜。)