提交 6259dba5 编写于 作者: J JiabinYang

test=develop

上级 eab2e5e5
......@@ -7470,10 +7470,11 @@ def reorg(x, stride, name=None):
x=data, stride=2)
"""
helper = LayerHelper("reorg", **locals())
if not (isinstance(stride, int)):
raise ValueError("stride must be a python Int")
helper = LayerHelper("reorg", **locals())
if name is None:
out = helper.create_tmp_variable(dtype=x.dtype)
else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册