发新帖

TensorFlow c++调用异常

[复制链接]
66 0

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
我使用GitHub上找到的c++动态库,1.10版本的,按照官方的代码,运行的时候爆出异常,说内存溢出,为什么啊,求教,
int main()
{
        using namespace tensorflow;
        using namespace tensorflow::ops;
        Scope root = Scope::NewRootScope();
        // Matrix A = [3 2; -1 0]
        auto A = Const(root, { { 3.f, 2.f },{ -1.f, 0.f } });
        //// Vector b = [3 5]
        auto b = Const(root, { { 3.f, 5.f } });
        //// v = Ab^T
        auto v = MatMul(root.WithOpName("v"), A, b, MatMul::TransposeB(true));
        std::vector<Tensor> outputs;
        ClientSession session(root);
       
        //// Run and fetch v
        TF_CHECK_OK(session.Run({ v }, &outputs));
        //// Expect outputs[0] == [19; -3]
        //LOG(INFO) << outputs[0].matrix<float>();
       
        return 0;
}

1{OODL4ETP~5I37%EY_9S}O.png
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表