mxnet基础到提高(24)-C++-符号,变量读写

    xiaoxiao2022-07-14  158

    #include #include #include “mxnet-cpp/MxNetCpp.h” //email:lx@lxaipro.com //code:liuxing using namespace std; using namespace mxnet::cpp; Symbol myNet(){ auto x=Symbol::Variable(“X”); auto y=Symbol::Variable(“Y”); auto z=Symbol::Variable(“Z”); z=x+y; return z; } int main(){ std::map<std::string, NDArray> args; Context ctx = Context::cpu(); auto dataX=NDArray(Shape(1), ctx); auto dataY=NDArray(Shape(1), ctx); dataX=12.5; dataY=22.9; args[“X”]=dataX; args[“Y”]=dataY; auto net=myNet(); auto *exe=net.SimpleBind(ctx,args); exe->Forward(false); std::vector& out = exe->outputs; cout<<out.size()<<endl; cout<<out[0]<<endl; delete exe; }

    $make learn1 $build/learn1

    1 [35.4, ]

    最新回复(0)