diff --git a/python/paddle/fluid/tests/unittests/test_lrn_op.py b/python/paddle/fluid/tests/unittests/test_lrn_op.py index 2b632b2437ad64841a48ee96b8de4f54c6889e21..588241969897f6936d9434411b42a032a5ca3990 100644 --- a/python/paddle/fluid/tests/unittests/test_lrn_op.py +++ b/python/paddle/fluid/tests/unittests/test_lrn_op.py @@ -323,6 +323,12 @@ class TestLocalResponseNormFAPIError(unittest.TestCase): self.assertRaises(ValueError, test_dim) + def test_shape(): + x = fluid.data(name='x', shape=[0, 0, 2, 3], dtype="float32") + paddle.nn.functional.local_response_norm(x, size=5) + + self.assertRaises(ValueError, test_shape) + class TestLocalResponseNormCAPI(unittest.TestCase): def setUp(self): diff --git a/python/paddle/nn/functional/norm.py b/python/paddle/nn/functional/norm.py index 863787c00e649dbadf1c29681e4652100210e969..c61fafdcb91549177e23520c91469878d0a71371 100644 --- a/python/paddle/nn/functional/norm.py +++ b/python/paddle/nn/functional/norm.py @@ -487,6 +487,12 @@ def local_response_norm(x, 'Expected 3D or higher dimensionality input, but got {} dimensions'. format(dim)) + for i, size in enumerate(sizes): + if not size > 0: + raise ValueError("Expected every dim's size to be larger than 0, " + "but the size of the {}-th dim is {}".format(i, + size)) + channel_last = True if data_format[-1] == "C" else False from functools import reduce