如何在TensorFlow C++；应用程序编程接口？ 我想用C++加载TysFooRoad模型。我想知道模型输入的大小，它是模型中的占位符

我用谷歌搜索了这个问题，但我只是在stackoverflow中找到了这个链接：

虽然我可以得到节点，但tensorflow文档没有告诉我如何访问节点的大小。有人知道这件事吗

---

非常感谢你

好的，经过多次尝试。我已经找到了一个解决办法，它可能很棘手，但效果很好

首先，我们可以使用以下代码获取占位符节点：

GraphDef mygd = graph_def.graph_def();
for (int i = 0; i < mygd.node_size(); i++)
{
    if (mygd.node(i).name() == input_name)
    {
        auto node = mygd.node(i);
    }
}

然后通过attr_value.cc（tensorflow/core/framework/attr_value.cc），我们可以通过如下代码获得形状attr值：

auto attr = node.attr();

tensorflow::AttrValue shape = attr["shape"];

[?,1024]

形状属性值是用于存储形状信息的结构。我们可以通过tensorflow/core/framework/attr\u value\u util.h中的函数summaryATTRValue获得详细信息

string size_summary = SummarizeAttrValue(shape);

然后我们可以得到如下形状的字符串格式：

auto attr = node.attr();

tensorflow::AttrValue shape = attr["shape"];

[?,1024]

---

谢谢你的支持。投票结果差一点让我失望。你知道怎么取尺寸吗？我在github中搜索tensorflow源代码，没有找到有用的答案。