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

test=develop

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