From a6fc250d6bdaf6d63698f7324ca6f27934cbcd34 Mon Sep 17 00:00:00 2001
From: zhongpu <2013000149@qq.com>
Date: Thu, 24 Oct 2019 14:33:25 +0800
Subject: [PATCH] remove assert statement to support sqeeze op in drgraph,
 test=release/1.6 (#20763) (#20801)

---
 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 48d04101fbd..a2f29fa02a0 100755
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -8979,8 +8979,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