提交 b00fbd96 编写于 作者: F fengjiayi

fix error

上级 d684b49c
...@@ -637,10 +637,10 @@ def read_file(reader): ...@@ -637,10 +637,10 @@ def read_file(reader):
out = [ out = [
helper.create_tmp_variable( helper.create_tmp_variable(
stop_gradient=True, dtype='float32') stop_gradient=True, dtype='float32')
for _ in range(len(file_obj.desc.shapes())) for _ in range(len(reader.desc.shapes()))
] ]
helper.append_op( helper.append_op(
type='read', inputs={'Reader': [file_obj]}, outputs={'Out': out}) type='read', inputs={'Reader': [reader]}, outputs={'Out': out})
if len(out) == 1: if len(out) == 1:
return out[0] return out[0]
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册