提交 3f815e07 编写于 作者: L liuwei1031 提交者: dzhwinter

fix github issue 15267 test=develop (#15306)

* fix github issue 15267 test=develop

* fix github issue 15267 test=develop
上级 0fbb76f6
......@@ -8479,8 +8479,7 @@ def shape(input):
"""
helper = LayerHelper('shape', **locals())
out = helper.create_variable_for_type_inference(
dtype=helper.input_dtype('input'))
out = helper.create_variable_for_type_inference(dtype='int32')
helper.append_op(
type='shape', inputs={'Input': input}, outputs={'Out': out})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册