c++ - 如何在 C++ API 中使用 NDArray？

Question

我正在更改 mxnet 中的 C++ 示例。我不明白如何分配 NDArray 对象。甚至没有基本的文档，这非常令人沮丧。

我尝试分配一个 NDArray，但是通过声明一个实例它似乎并没有分配数据，只有当我用数据填充一个数组时。对吗？

// this code snippet does not work     
  NDArray a = NDArray(Shape(10, 20), Context::cpu());
  const float *dat = a.GetData();
  float result = *dat; // read memory violation
  result = *(dat + 10);

// this code snippet works
  NDArray b = NDArray(Shape(10, 20), Context::cpu());
  a.SampleUniform(1.0, 2.0, &b);
  const float *dat2 = b.GetData();
  float result2 = *dat2; // works!!
  result2 = *(dat2 + 10); 

有人使用过 C++ API 和不断变化的网络吗？

Answer 1

还有第三个参数delay_alloc: https://github.com/apache/incubator-mxnet/blob/master/cpp-package/include/mxnet-cpp/ndarray.h#L144

将其设置为 false，您的代码将起作用。