Fix iterator over symbol when multiple children have the same name (a…

…pache#14597)
haohuanw · Jun 23, 2019 · 81c2425 · 81c2425
1 parent 09c50e7
commit 81c2425
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/python/mxnet/symbol/symbol.py b/python/mxnet/symbol/symbol.py
@@ -90,7 +90,7 @@ def __iter__(self):
         <Symbol d>
         <Symbol _plus0>
         """
-        return (self[i] for i in self.list_outputs())
+        return (self[i] for i in range(len(self)))
 
     def __add__(self, other):
         """x.__add__(y) <=> x+y

diff --git a/tests/python/unittest/test_symbol.py b/tests/python/unittest/test_symbol.py
@@ -367,6 +367,11 @@ def test_simple_bind_gradient_graph_possible_with_cycle():
     res = data + data + data + data + data + data + data + data
     res.simple_bind(ctx=mx.cpu(), data=(1,))
 
+def test_children_same_name():
+    a = mx.sym.Variable('data')
+    b = a + a
+    for c in b.get_children():
+        pass
 
 if __name__ == '__main__':
     import nose