diff --git a/python/paddle/nn/functional/common.py b/python/paddle/nn/functional/common.py index 7319b860db8f79262c6f5ab307bed15145472ce7..fac5ca2f7936ee1e4f704bad03cbca4a938d1494 100644 --- a/python/paddle/nn/functional/common.py +++ b/python/paddle/nn/functional/common.py @@ -1276,6 +1276,9 @@ def pad(x, pad, mode='constant', value=0, data_format="NCHW", name=None): x_dim = len(x.shape) + if mode == "constant" and isinstance(pad, list) and len(pad) == x_dim * 2: + return layers.pad(x, pad, pad_value=value) + assert x_dim in [ 3, 4, 5 ], "input tesor dimension must be in [3, 4, 5] but got {}".format(x_dim) @@ -1291,9 +1294,6 @@ def pad(x, pad, mode='constant', value=0, data_format="NCHW", name=None): unsqueezed_dim = [] - if mode == "constant" and isinstance(pad, list) and len(pad) == x_dim * 2: - return layers.pad(x, pad, pad_value=value) - if isinstance(pad, Variable): if data_format in ["NCL", "NCHW", "NCDHW"]: data_format = "NCDHW"