提交 4a8e399f 编写于 作者: Y Yibing Liu

Add back one wrong removed function

上级 44facf18
......@@ -35,6 +35,13 @@ def lodtensor_to_ndarray(lod_tensor):
return ret, lod_tensor.lod()
def split_infer_result(infer_seq, lod):
infer_batch = []
for i in xrange(0, len(lod[0]) - 1):
infer_batch.append(infer_seq[lod[0][i]:lod[0][i + 1]])
return infer_batch
class CriticalException(Exception):
pass
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册