【问题标题】:How to fill the tensor in tensorflow c++?如何在tensorflow c ++中填充张量?
【发布时间】: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


【解决方案1】:

其实上面提到的方法是在tensorflow中填充张量比较好的方法

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2020-01-13
    • 2018-10-14
    • 1970-01-01
    • 2018-03-24
    • 1970-01-01
    • 2016-03-12
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多