test=develop

6259dba5 · JiabinYang · eab2e5e5 · 6259dba5
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +2 -1

未找到文件。
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -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: