diff --git a/compressai/entropy_models/entropy_models.py b/compressai/entropy_models/entropy_models.py
index acb98333..58501e8c 100644
--- a/compressai/entropy_models/entropy_models.py
+++ b/compressai/entropy_models/entropy_models.py
@@ -158,7 +158,7 @@ def quantize(
 
         if mode == "noise":
             half = float(0.5)
-            noise = torch.empty_like(inputs).uniform_(-half, half)
+            noise = 2*half*torch.rand_like(inputs) - half
             inputs = inputs + noise
             return inputs