c++ - 从 std::vector 创建 tensorflow::tensor 的最有效方法

Question

所以我的问题是知道是否有一种方法可以直接将值从 vector(但我们也可以考虑 array)传递给 tensorflow ::张量?

我知道的唯一方法是一个一个地复制每个值。

示例(二维 vector ):

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];
  }
}

有没有更方便的方法呢？

例如array到vector:

int x[3] = {1, 2, 3};
std::vector<int> v(x, x + sizeof x / sizeof x[0]);

Answer 1

这个怎么样？ std::copy_n(vec.begin(), vec.size(), input.flat<float>().data())