Added eps

TorchDSP · Sep 18, 2023 · ee4487a · ee4487a
1 parent a4e97fe
commit ee4487a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torchsig/transforms/functional.py b/torchsig/transforms/functional.py
@@ -1026,7 +1026,10 @@ def quantize(
 
     """
     # Setup quantization resolution/bins
-    max_value = max(np.abs(np.real(np.copy(tensor)),np.imag(np.copy(tensor)))) + 1e-9
+    max_value = (
+        max(np.abs(np.real(np.copy(tensor)), np.imag(np.copy(tensor))))
+        + np.finfo(np.dtype(np.float64)).eps
+    )
     bins = np.linspace(-max_value, max_value, num_levels + 1)
 
     # Digitize to bins