Как заполнить тензор в C++

Question

Я создаю тензор так:

tensorflow::Tensor a(tensorflow::DT_FLOAT, tensorflow::TensorShape()); 

Я знаю, как заполнить скалярное значение:

a.scalar<float>()() = 8.0; 

Но я не знаю, как заполните тензор, подобный [1, 4, 2].

Answer 1

Есть несколько вариантов. Если тензор действительно небольшой вектор, как в вашем случае, вы можете сделать следующее:

tensorflow::Tensor a(tensorflow::DT_FLOAT, tensorflow::TensorShape(3)); 
a.vec<float>()(0) = 1.0f; 
a.vec<float>()(1) = 4.0f; 
a.vec<float>()(2) = 2.0f; 

Если вы хотите, чтобы построить немного больше и/или многомерный тензор, то tensorflow::ops::Input::Initializer объявлен в tensorflow/cc/framework/ops.h имеет много конструкторы, которые позволяют вам построить Tensor из различных типов констант C++, таких как простые примитивные константы и вложенные списки инициализаторов, представляющие многомерный массив.

Например, если вы хотите построить 2x2 матрицу, вы можете сделать следующее:

#include "tensorflow/cc/framework/cc/ops.h" 

tensorflow::ops::Input::Initializer a({{1, 2}, {3, 4}}); 
// a.tensor will be a Tensor with type DT_INT32 and shape {2, 2}.