Update i6_models/parts/conformer/mhsa_rel_pos.py

Co-authored-by: Albert Zeyer <[email protected]>

i6_models/parts/conformer/mhsa_rel_pos.py

@@ -144,7 +144,7 @@ def forward(self, input_tensor: torch.Tensor, sequence_mask: torch.Tensor) -> to
         mask = (
             torch.zeros_like(inv_sequence_mask, dtype=input_tensor.dtype)
             .masked_fill(inv_sequence_mask, float("-inf"))
-            .view(batch_dim_size, 1, 1, time_dim_size)
+            .reshape(batch_dim_size, 1, 1, time_dim_size)
         )  # [B, 1, 1, T']
 
         # query, key and value sequences