c++ - TensorFlow C++ 中用于 feed_dict 的语法是什么？

Question

我想在 TensorFlow C++ 中构建和训练一个由两层组成的图，并将给定的矩阵作为输入提供给它。

我有两个不同的语法示例:

The official C++ example (line # 129)

An old answer in StackOverflow

关于 tensorflow::Session::Run() 的“输入”参数的确切语法，它们似乎相互矛盾

应该是“placeholder_name:0”还是“placeholder_name”？

Answer 1

两者皆可。名称通过 ParseTensorName 传递，其中假定没有冒号的名称的输出索引为 0。为了验证这一点，我们可以在 DirectSessionMinusAXTest::TestFeed 中的提要名称末尾添加一个“:0”。 :

std::vector<std::pair<string, Tensor>> inputs = {{x_, t}};

成为

std::vector<std::pair<string, Tensor>> inputs = {{x_ + ":0", t}};

它仍然过去了。

唯一需要传递输出索引的情况(更准确地说是应该需要它的唯一情况；可能有一些代码缺乏规范化)是如果你正在提供一个不是操作的第零输出的张量(例如“唯一:1”)。这是非常罕见的，因为常量和占位符操作是最有可能的提要目标并且只有一个输出。