【发布时间】:2021-02-20 09:13:17
【问题描述】:
我正在用 tensorflow c++ 加载 .pb 文件。现在我需要用我的数据填充输入张量并获取输出张量。
为了填充数据,我使用以下代码:
tensorflow::Tensor points_tensor{tensorflow::DataType::DT_FLOAT, tensorflow::TensorShape({number_of_points,4})};
auto pointsMapped = points_tensor.tensor<float, 2>();
for(int i=0; i<number_of_points; i++){
//to the shifting here only
pointsMapped(i,0) = point_cloud.points[i].x;
pointsMapped(i,1) = point_cloud.points[i].y;
pointsMapped(i,2) = point_cloud.points[i].z;
pointsMapped(i,3) = point_cloud.points[i].intensity;
}
point_cloud 是点对象的向量。
但我不这么认为,这是用 c++ 做的更好的方法。因为我需要访问我的张量形状。
有人可以帮我解决这个问题吗?
【问题讨论】:
-
@CppProgrammer23 我已经检查过了,但我无法弄清楚如何用那个答案填充张量。你能帮我吗?
-
你不明白什么?你能指出一个具体的点以便我解释吗?
-
@CppProgrammer23 在链接中,它们正在填充一维张量,例如 a.vec
()(0)。但我需要填充 n-d 张量作为示例(200,4)。我不能使用上面的代码行来做到这一点。我只需要知道在张量流中填充张量的签名是什么 -
我找到了这个:tensorflow.org/api_docs/cc/class/tensorflow/ops/fill这对你有帮助吗?
标签: c++ tensorflow