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

… == data_out->ndim (2 vs. 1) (apache#1987)
eqy · Oct 29, 2018 · 6ceaca5 · 6ceaca5
1 parent e23e15c
commit 6ceaca5
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)