时间戳

首页 > 解决方案 > 如何在tensorflow c ++中填充张量？

c++ - 如何在tensorflow c ++中填充张量？

问题描述

我正在用 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++ 做的更好的方法。因为我需要访问我的张量形状。

有人可以帮我弄这个吗？

标签： c++tensorflow

解决方案

实际上上面提到的方法是在tensorflow中填充张量的更好方法

推荐阅读