fix masked scatter

rwth-i6 · Jan 20, 2025 · f17a530 · f17a530
1 parent 04fc206
commit f17a530
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/users/zeyer/experiments/exp2024_04_23_baselines/recog_ext/ctc.py b/users/zeyer/experiments/exp2024_04_23_baselines/recog_ext/ctc.py
@@ -415,8 +415,6 @@ def _masked_scatter(
     merged_dim_map: Dict[Dim, Dim],
 ) -> T:
     if isinstance(s, Tensor):
-        if in_dim not in s.dims:
-            return s  # e.g. scalar or so, independent from masking
         assert isinstance(backup, Tensor)
         # Do the reverse of _masked_select above.
         # First replace the dims back.