所以我的问题是要知道是否有一种方法可以直接将向量中的值(但我们也可以考虑数组)传递给tensorflow :: tensor?
我知道的唯一方法是逐个复制每个值.
示例(2D矢量):
tensorflow::Tensor input(tensorflow::DT_FLOAT, tensorflow::TensorShape({50, 20}));
auto input_map = input.tensor<float, 2>();
for (int b = 0; b < 50; b++) {
for (int c = 0; c < 20; c++) {
input_map(b, c) = (vector_name)[b][c];
}
}
有更方便的方法吗?
例如,数组到矢量:
int x[3] = {1, 2, 3};
std::vector<int> v(x, x + sizeof x / sizeof x[0]);
解决方法:
这个怎么样?
std :: copy_n(vec.begin(),vec.size(),input.flat< float>().data())