diff --git a/python/paddle/v2/inference.py b/python/paddle/v2/inference.py
index 32636c5505680471c1c8c300c2510f3823c38e8e..31a5d26e6ebc07a0105e02fd8fd5cc8181d5424e 100644
--- a/python/paddle/v2/inference.py
+++ b/python/paddle/v2/inference.py
@@ -37,16 +37,15 @@ class Inference(object):
         self.__gradient_machine__.finish()
 
     def iter_infer_field(self, field, **kwargs):
-        for result in self.iter_infer(**kwargs):
-            yield [
-                each_result[each_field]
-                for each_result in result for each_field in field
-            ]
-
-    def infer(self, field='value', **kwargs):
         if not isinstance(field, list) and not isinstance(field, tuple):
             field = [field]
 
+        for result in self.iter_infer(**kwargs):
+            for each_result in result:
+                item = [each_result[each_field] for each_field in field]
+                yield item
+
+    def infer(self, field='value', **kwargs):
         retv = None
         for result in self.iter_infer_field(field=field, **kwargs):
             if retv is None: