fix github issue 15267 test=develop (#15306)

* fix github issue 15267 test=develop * fix github issue 15267 test=develop
6 years ago · 3f815e079f
parent 0fbb76f66b
commit 3f815e079f
1 changed files with 1 additions and 2 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -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})