mapping: allow demapping infinite/high SNR symbols

Signed-off-by: japm48 <[email protected]>

sionna/mapping.py

@@ -980,6 +980,8 @@ def __init__(self,
                                               dtype.real_dtype,
                                               **kwargs)
 
+        self._no_threshold = np.finfo(dtype.as_numpy_dtype).tiny
+
     @property
     def constellation(self):
         return self._constellation
@@ -1001,6 +1003,8 @@ def call(self, inputs):
         # Add a dummy dimension for broadcasting. This is not needed when no
         # is a scalar, but also does not do any harm.
         no = tf.expand_dims(no, axis=-1)
+        # Deal with zero or very small values.
+        no = tf.math.maximum(no, self._no_threshold)
 
         # Compute exponents
         exponents = -squared_dist/no