Tensorflow:在C ++

时间:2018-12-03 06:57:43

标签: c++ tensorflow type-conversion

我有一些操作已经处理过:

auto net1 = ops::ReadFile(scope, filename);
auto net2 = ops::DecodePng(scope, net1);

std::vector<Tensor> outputs1;
session.Run({ net2 }, &outputs1);

现在我必须使用output向量作为新操作集的输入:

auto net3 = ?????
auto net4 = ops::Cast(scope, net3, DT_FLOAT);

std::vector<Tensor> outputs2;
session.Run({ net4 }, &outputs2);

问题是:如何将std :: vector转换为tensorflow :: Input?有任何内置方法吗?

1 个答案:

答案 0 :(得分:0)

我最终使用Tensor作为输入:

auto net3 = outputs1[0]

这对我来说可以,因为我只有一张图像。