fix github issue 15267 test=develop

b758fa50 · hjchen2 · e33427da · b758fa50
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +1 -2

未找到文件。
--- 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})