remove assert statement to support sqeeze op in drgraph, test=develop (#20763)

revert-20712-fix_depthwise_conv
zhongpu 6 years ago committed by Jiabin Yang
parent 67b59ddb38
commit 702aad5a0a

@ -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):

Loading…
Cancel
Save