From 702aad5a0ab1ee6550208bcbb99464d8dc84e4a1 Mon Sep 17 00:00:00 2001
From: zhongpu <2013000149@qq.com>
Date: Tue, 22 Oct 2019 14:48:29 +0800
Subject: [PATCH] remove assert statement to support sqeeze op in drgraph,
 test=develop (#20763)

---
 python/paddle/fluid/layers/nn.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index dd657d2c1d..2d12c7f3d4 100755
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -8952,8 +8952,6 @@ def squeeze(input, axes, name=None):
             y = layers.squeeze(input=x, axes=[2]) # y.shape=[None, 5, 10]
 
     """
-    assert not in_dygraph_mode(), (
-        "squeeze layer is not supported in dygraph mode yet.")
     helper = LayerHelper("squeeze", **locals())
 
     if not isinstance(input, Variable):
-- 
GitLab