Multiple output support in gluon backendrep

apache · Dec 26, 2018 · dc01588 · dc01588
1 parent 499f2fc
commit dc01588
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/tests/python-pytest/onnx/backend_rep.py b/tests/python-pytest/onnx/backend_rep.py
@@ -126,7 +126,12 @@ def run(self, inputs, **kwargs):
         net_inputs = [nd.array(input_data, ctx=ctx) for input_data in inputs]
         net_outputs = self.net(*net_inputs)
         results = []
-        results.extend([o for o in net_outputs.asnumpy()])
-        result = np.array(results)
+        if isinstance(net_outputs, list):
+            for output in net_outputs:
+                results.append(output.asnumpy())
+            result = results
+        else:
+            results.extend([o for o in net_outputs.asnumpy()])
+            result = [np.array(results)]
 
-        return [result]
+        return result