在Tensorflow C ++中创建多维张量

时间:2018-08-31 13:12:39

标签: c++ tensorflow

我正在尝试在优化中使用c ++中的tensorflow。我需要在tensorflow会话中运行的值是连续生成的,并且不遵循从图像文件格式进行的典型读取,因此我没有遵循典型的管道。我尝试从此链接使用下面的代码块 most efficient way to create tensorflow::tensor from std::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];
  }
}

但是我收到一个错误消息“没有重载函数类型tensorflow :: Tensor :: tensor的实例,当我进入API时,我也没有看到它的实例。阅读API时,我只看到了如何声明张量,而不是如何访问张量元素或分配值。由于平台要求,我正在使用Tensorflow 1.4版

理想情况下,我希望直接创建一个由float值填充的4-d张量,该张量表示要优化的问题。在这方面的任何帮助将不胜感激。谢谢。

0 个答案:

没有答案