Skip to content

Commit 2cb12fd

Browse files
authored
Fix data type issue in _compute_doppler_shifts, fixes #426 (#436)
Signed-off-by: Florian Euchner <florian.euchner@gmail.com> Merged-by: Guillermo Marcus <4169784+gmarcusm@users.noreply.github.com>
1 parent 8ad32bc commit 2cb12fd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sionna/rt/solver_paths.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4124,7 +4124,7 @@ def _compute_doppler_shifts(self, paths, paths_tmp, velocity):
41244124
# Compute Doppler shift per path
41254125
#[num_targets, num_sources, max_num_paths]
41264126
doppler = tf.reduce_sum(velocity*k_diff, axis=-1)
4127-
doppler = tf.where(objects_mask, 0, doppler)
4127+
doppler = tf.where(objects_mask, tf.constant(0, doppler.dtype), doppler)
41284128
doppler = tf.reduce_sum(doppler, axis=0)
41294129
doppler /= self._scene.wavelength
41304130
return doppler

0 commit comments

Comments
 (0)