在Tensorflow C ++中连接多个张量

时间:2019-08-05 23:29:00

标签: c++ tensorflow

我想将多个tensorflow :: Tensor连接在一起,以仅使用一次会话,我有一个过程可以为每个图像检索张量,并使用此代码建议import-opencv-mat-into-c-tensorflow-without-copying

填充数据。

但是我在尝试合并所有这些张量时遇到问题,并且函数concat的文档尚不清楚如何执行此操作,此刻我的代码如下:

std::vector<tensorflow::Tensor> inputTensorVector;
for(cv::Mat image: images){
    inputTensorVector.push_back(create_vector(image));
}

tensorflow::Tensor input(tensorflow::DT_FLOAT, tensorflow::TensorShape({ inputTensorVector.size(), 100, 100, 3 }));

create_vector是一个从作为参数传递的图像中创建形状为(1,100,100,3)的张量的函数,问题是如何用inputTensorVector中的每个张量填充输入张量?

0 个答案:

没有答案