diff --git a/python/paddle/fluid/tests/unittests/test_randint_op.py b/python/paddle/fluid/tests/unittests/test_randint_op.py index 89739a37fd95b1eceb9a1899086975e3a03e98a7..5b2d5be346a9b205cb44373f58a413baa6c8a2fa 100644 --- a/python/paddle/fluid/tests/unittests/test_randint_op.py +++ b/python/paddle/fluid/tests/unittests/test_randint_op.py @@ -57,6 +57,7 @@ class TestRandintOpError(unittest.TestCase): self.assertRaises(TypeError, paddle.randint, 5, shape=np.array([2])) self.assertRaises(TypeError, paddle.randint, 5, dtype='float32') self.assertRaises(ValueError, paddle.randint, 5, 5) + self.assertRaises(ValueError, paddle.randint, -5) class TestRandintOp_attr_tensorlist(OpTest): diff --git a/python/paddle/tensor/random.py b/python/paddle/tensor/random.py index 8ef9dde0880795c08342d95d0f80cd2ea6e2b6dc..eac99163e05794db4a6afdb529a551d960ee3c02 100644 --- a/python/paddle/tensor/random.py +++ b/python/paddle/tensor/random.py @@ -114,6 +114,10 @@ def randint(low=0, high=None, shape=[1], dtype=None, name=None): """ if high is None: + if low <= 0: + raise ValueError( + "If high is None, low must be greater than 0, but received low = {0}.". + format(low)) high = low low = 0 if dtype is None: