機械学習AIのmxnet c++エッセイ(4)
1094 ワード
#include
#include
#include "mxnet-cpp/MxNetCpp.h"
//email:[email protected]
//code:liuxing
using namespace std;
using namespace mxnet::cpp;
int main(){
Context ctxDev(DeviceType::kCPU, 0);
Context ctx = Context::cpu();
std::vector inArgs;
std::vector argGradStore;
std::vector gradReqType;
std::vector 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& out = exe->outputs;
cout<
$make learn1 $build/learn1 1 [21.58, ]