Tensorflow C++ YOU_MADE_A_PROGRAMMING_MISTAKE

Question

我用python训练我的NN模型，并在VS2015 C++中加载它。这段代码：

    // The session will initialize the outputs
    vector<Tensor> outputs;

    // Run the session, evaluating our "c" operation from the graph
    status = session->Run(inputs, { "y_pred" }, {}, &outputs);

    // Convert the node to a scalar representation.
    auto output_c = outputs[0].flat<float>();

y_pred是一个二元张量，所以我用平面来得到它。但是，我从EIGEN_STATIC_ASSERT那里得到了一个错误，“EIGEN_STATIC_ASSERT”。

以前有人有这个问题吗？我该怎么解决呢？谢谢!

Answer 1

最后，我在堆栈溢出中找到了一篇文章，但我无法确定谁是原作者。实际上，我们需要平面功能。

session->Run(inputs, { "pred" }, {}, &outputs);
TTypes<float>::Flat indices_flat = outputs[0].flat<float>();
float coutput[6];
for (int i = 0; i<dataSize; i++)    { 
    coutput[i] = indices_flat(i);
    cout << "outptut[i]: " << indices_flat(i) << endl; 
}