Fix the unstack layer (#15047)

test=develop

Fix the unstack layer (#15047)
test=develop
a28df3eb · Yibing Liu · GitHub · b9fb03cf · a28df3eb
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +1 -1

未找到文件。
--- 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(