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

remove assert statement to support sqeeze op in drgraph, test=release/1.6 (#20763) (#20801)

上级 a30803eb
...@@ -8979,8 +8979,6 @@ def squeeze(input, axes, name=None): ...@@ -8979,8 +8979,6 @@ def squeeze(input, axes, name=None):
y = layers.squeeze(input=x, axes=[2]) # y.shape=[None, 5, 10] 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()) helper = LayerHelper("squeeze", **locals())
if not isinstance(input, Variable): if not isinstance(input, Variable):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册