[DOCS] Fix C++ example:graph_runtime.cc:151: Check failed: data->ndim…

… == data_out->ndim (2 vs. 1) (apache#1987)
neo-ai · Feb 20, 2019 · 43c5140 · 43c5140
1 parent 3547120
commit 43c5140
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/deploy/nnvm.md b/docs/deploy/nnvm.md
@@ -96,8 +96,8 @@ int main()
     run();
 
     DLTensor* y;
-    int out_ndim = 1;
-    int64_t out_shape[1] = {1000, };
+    int out_ndim = 2;
+    int64_t out_shape[2] = {1, 1000, };
     TVMArrayAlloc(out_shape, out_ndim, dtype_code, dtype_bits, dtype_lanes, device_type, device_id, &y);
 
     // get the function from the module(get output data)