From 3f815e079fa6eaa9ce4d00a4de65eac922c6e873 Mon Sep 17 00:00:00 2001
From: liuwei1031 <46661762+liuwei1031@users.noreply.github.com>
Date: Tue, 15 Jan 2019 11:40:48 +0800
Subject: [PATCH] fix github issue 15267 test=develop (#15306)

* fix github issue 15267 test=develop

* 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