From 88881955e729596bf916bc8382df8fd8b5bc8e0a Mon Sep 17 00:00:00 2001 From: "liuwei(DLTP)" Date: Mon, 14 Jan 2019 10:24:18 +0800 Subject: [PATCH] fix github issue 15267 test=develop --- python/paddle/fluid/layers/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index a4787e769f..99e1c2adfd 100644 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -8480,7 +8480,7 @@ def shape(input): helper = LayerHelper('shape', **locals()) out = helper.create_variable_for_type_inference( - dtype=helper.input_dtype('input')) + dtype='int32') helper.append_op( type='shape', inputs={'Input': input}, outputs={'Out': out}) -- GitLab