fix masked scatter

rwth-i6 · Jan 20, 2025 · 6ed21f5 · 6ed21f5
1 parent 21a3413
commit 6ed21f5
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
@@ -418,8 +418,6 @@ def _masked_scatter(
         assert isinstance(backup, Tensor)
         if in_dim not in s.dims:
             s = rf.expand_dim(s, in_dim)
-        if in_dim not in backup.dims:
-            backup = rf.expand_dim(backup, in_dim)
         # Do the reverse of _masked_select above.
         # First replace the dims back.
         if any(d in reverse_dim_map for d in s.dims):