提交 702aad5a 编写于 作者: Z zhongpu 提交者: Jiabin Yang

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

上级 67b59ddb
......@@ -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):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册