diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index dd657d2c1d89472d4448bedcd9e3a561513778ff..2d12c7f3d490e2401c268537e33b339d2c49ba0a 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):