Fix the unstack layer (#15047)

test=develop
6 years ago · a28df3eb0b
parent b9fb03cf54
commit a28df3eb0b
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -7943,7 +7943,7 @@ def unstack(x, axis=0, num=None):
            num = x.shape[axis]

    outs = []
-    for _ in num:
+    for _ in range(num):
        outs.append(helper.create_variable_for_type_inference(x.dtype))

    helper.append_op(