From 4bb40baae09e4baf637ae2673d156a28546a2d0c Mon Sep 17 00:00:00 2001 From: Howave Date: Fri, 26 Oct 2018 00:29:00 +0800 Subject: [PATCH] [DOCS] Fix C++ example:graph_runtime.cc:151: Check failed: data->ndim == data_out->ndim (2 vs. 1) (#1987) --- docs/deploy/nnvm.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/deploy/nnvm.md b/docs/deploy/nnvm.md index e4ce14528b53..1e0d17f8b195 100644 --- 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)