给像素矢量作为张量流的输入

Question

我在T ++中使用Tensorflow来解决图像分类问题。

我创建了图表并尝试使用示例代码here。

当我将图像（.jpeg）作为主函数的输入（主函数中的string image）时，它可以正常工作。但是我在2d向量std::vector<std::vector<int>> vec2d中具有图像的像素值（仅亮度值）。如何将此向量作为进行预测的输入？

我得到了一个如下所示的张量，但我无法弄清楚如何将其与现有代码相匹配。

tensorflow::Tensor input(tensorflow::DT_FLOAT, 
tensorflow::TensorShape({32, 32}));
auto input_map = input.tensor<float, 2>();


for (int b = 0; b < 32; b++) {
    for (int c = 0; c < 32; c++) {
      input_map(b, c) = (vec2d)[b][c];
    }
}

或者是否有内置的方法来传递张量流中的像素值？

我不想用像素值创建图像并重新读取它。我已经做了并且它有效但文件读/写操作需要时间，而且在我的系统中时间非常重要。