From cadabe1d7720949a1a45b6b85378b8fd50013724 Mon Sep 17 00:00:00 2001 From: Tao Luo Date: Thu, 13 Apr 2017 10:31:46 +0800 Subject: [PATCH] Update inference.py --- python/paddle/v2/inference.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/python/paddle/v2/inference.py b/python/paddle/v2/inference.py index 32636c5505..31a5d26e6e 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: -- GitLab