Как я могу получить данные из тензорной переменной в C ++

Question

Мне очень хотелось бы узнать, как я могу получить данные из объекта тензорного потока Variabe с использованием API C ++. Например, для чистого объекта Tensor со ссылкой, например, obj, выполнение obj->flat<Type>().data() возвращает все тензорные данные. Есть ли аналогичный способ извлечения данных из объекта Variable?

Answer 1

Только что получил подсказку, как это сделать. Я запускаю переменную в сеансе и отправляю результаты в выходной вектор Tensor. Затем я извлекаю значения из соответствующего объекта Tensor, как описано в вопросе.

std::vector<Output> outputs;
auto my_var = Variable (...);
TF_CHECK_OK(session.Run({my_var},&outputs));
print(output[0]);