From b758fa50b2155121f94b043967eb36ebb0c87cf6 Mon Sep 17 00:00:00 2001 From: hjchen2 Date: Mon, 14 Jan 2019 11:09:27 +0800 Subject: [PATCH] fix github issue 15267 test=develop --- python/paddle/fluid/layers/nn.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index a4787e769..56971cff4 100644 --- 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}) -- GitLab