diff --git a/sionna/rt/solver_paths.py b/sionna/rt/solver_paths.py index d7eff10..5561ca2 100644 --- 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