如何在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，你的代碼將工作。