diff --git a/python/paddle/v2/fluid/io.py b/python/paddle/v2/fluid/io.py index 57f051985ca0b91a071ad2ef27c48a00b0b615ef..e7a06a07145758335986368fbe111de2bb930246 100644 --- a/python/paddle/v2/fluid/io.py +++ b/python/paddle/v2/fluid/io.py @@ -198,10 +198,11 @@ def prepend_feed_ops(inference_program, feeded_var_names): name='feed', type=core.VarDesc.VarType.FEED_MINIBATCH, persistable=True) for i, name in enumerate(feeded_var_names): + out = global_block.var(name) global_block.prepend_op( type='feed', inputs={'X': [feed_var]}, - outputs={'Out': [name]}, + outputs={'Out': [out]}, attrs={'col': i})