fix

apache · Dec 6, 2018 · 24e8ac3 · 24e8ac3
1 parent c4d129b
commit 24e8ac3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/contrib/onnx/onnx2mx/import_onnx.py b/python/mxnet/contrib/onnx/onnx2mx/import_onnx.py
@@ -201,7 +201,7 @@ def _parse_array(self, tensor_proto):
             np_array = to_array(tensor_proto).reshape(tuple(tensor_proto.dims))
         else:
             np_array = np.array([to_array(tensor_proto)])
-            return nd.array(np_array)
+        return nd.array(np_array)
 
     def _parse_attr(self, attr_proto):
         """Convert a list of AttributeProto to a dict, with names as keys."""