Advertisement
Guest User

Untitled

a guest
Jun 26th, 2017
68
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.63 KB | None | 0 0
  1.     Symbol a = Symbol::Variable("a");
  2.     Symbol b = Symbol::Variable("b");
  3.     b = elemwise_add("added_b", a, b);
  4.  
  5.     vector<mx_float> vA, vB;
  6.     vA.push_back(3.);
  7.     vB.push_back(5.);
  8.     auto tmpA = NDArray(vA.data(), Shape(1), ctx_cpu);
  9.     auto tmpB = NDArray(vB.data(), Shape(1), ctx_cpu);
  10.     args_map["a"] = tmpA;
  11.     args_map["b"] = tmpB;
  12.     Executor *exe = b.SimpleBind(ctx_cpu, args_map);
  13.     exe->Forward(false);
  14.  
  15.     const auto &out = exe->outputs;
  16.     NDArray out_cpu = out[0].Copy(ctx_cpu);
  17.     NDArray::WaitAll();
  18.     const mx_float *dptr_out = out_cpu.GetData();
  19.     fprintf(stderr, "b = %f\n", dptr_out[0]);
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement