如何从 C++ 中的 tensorflow 变量中获取数据

Question

我很想知道如何使用 C++ ZDB974238714CA8DE63487A 从 tensorflow Variabe object 获取数据。 例如，对于带有引用 obj 的纯张量 object，执行obj->flat<Type>().data()会返回所有张量数据。 是否有类似的方法从变量 object 中提取数据？

Answer 1

刚刚得到了关于如何做到这一点的提示。 我在 session 中运行变量并将结果发送到 output 张量向量。 然后我从相应的张量 object 中提取值，如问题中所述。

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