Fix a crash in android_deploy demo. (apache#2073)

neo-ai · Feb 20, 2019 · 3677708 · 3677708
1 parent 3b343b4
commit 3677708
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apps/android_deploy/app/src/main/java/ml/dmlc/tvm/android/demo/MainActivity.java b/apps/android_deploy/app/src/main/java/ml/dmlc/tvm/android/demo/MainActivity.java
@@ -298,7 +298,7 @@ protected Integer doInBackground(Bitmap... bitmaps) {
 
                     // get the function from the module(get output data)
                     Log.i(TAG, "get output data");
-                    NDArray outputNdArray = NDArray.empty(new long[]{1000}, new TVMType("float32"));
+                    NDArray outputNdArray = NDArray.empty(new long[]{1, 1000}, new TVMType("float32"));
                     Function getOutputFunc = graphRuntimeModule.getFunction("get_output");
                     getOutputFunc.pushArg(OUTPUT_INDEX).pushArg(outputNdArray).invoke();
                     float[] output = outputNdArray.asFloatArray();
@@ -630,4 +630,4 @@ public static Matrix getTransformationMatrix(
 
         return matrix;
     }
-}
+}