Fix the output of conditional_block

8 years ago · 787d1d36d0
parent 564dba1753
commit 787d1d36d0
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/v2/fluid/layers/control_flow.py
+++ b/python/paddle/v2/fluid/layers/control_flow.py
@ -897,7 +897,7 @@ class ConditionalBlock(object):

        out_list = [
            parent_block.var(var_name) for var_name in parent_block.vars
-            if var_name not in intermediate
+            if var_name in intermediate
        ]

        step_scope = parent_block.create_var(