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

所以我的问题是要知道是否有一种方法可以直接将向量中的值(但我们也可以考虑数组)传递给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())