mxnet基础到提高(23)-C++-数据读入与写出

    xiaoxiao2022-07-14  164

    #include <iostream> #include <vector> #include "mxnet-cpp/MxNetCpp.h" //email:lx@lxaipro.com //code:liuxing using namespace std; using namespace mxnet::cpp; int main(){ Context ctxDev(DeviceType::kCPU, 0); Context ctx = Context::cpu(); std::vector<NDArray> inArgs; std::vector<NDArray> argGradStore; std::vector<OpReqType> gradReqType; std::vector<NDArray> auxStates; NDArray arrayY(Shape(1), ctxDev, false); Symbol y("Y"); auto z=y+9; //---- mx_float* aptrY = new mx_float[1]; aptrY[0]=12.58; arrayY.SyncCopyFromCPU(aptrY,1); arrayY.WaitToRead(); inArgs.push_back(arrayY); argGradStore.push_back(NDArray()); gradReqType.push_back(kNullOp); Executor* exe = new Executor(z, ctxDev, inArgs, argGradStore,gradReqType,auxStates); exe->Forward(false); std::vector<NDArray>& out = exe->outputs; cout<<out.size()<<endl; cout<<out[0]<<endl; delete exe; }

    $make learn1 $build/learn1 1 [21.58, ]

    最新回复(0)