Fix data type issue in _compute_doppler_shifts, fixes NVlabs#426

Signed-off-by: Florian Euchner <florian.euchner@gmail.com>
Jeija · May 10, 2024 · a794225 · a794225
1 parent 8ad32bc
commit a794225
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sionna/rt/solver_paths.py b/sionna/rt/solver_paths.py
@@ -4124,7 +4124,7 @@ def _compute_doppler_shifts(self, paths, paths_tmp, velocity):
         # Compute Doppler shift per path
         #[num_targets, num_sources, max_num_paths]
         doppler = tf.reduce_sum(velocity*k_diff, axis=-1)
-        doppler = tf.where(objects_mask, 0, doppler)
+        doppler = tf.where(objects_mask, tf.constant(0, doppler.dtype), doppler)
         doppler = tf.reduce_sum(doppler, axis=0)
         doppler /= self._scene.wavelength
         return doppler