[Fix] type promotion (#3817)

PaddlePaddle · Oct 30, 2024 · 6316514 · 6316514
1 parent 658d19a
commit 6316514
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlespeech/audio/utils/tensor_utils.py b/paddlespeech/audio/utils/tensor_utils.py
@@ -248,7 +248,7 @@ def st_reverse_pad_list(ys_pad: paddle.Tensor,
     #   >>> tensor([[ 2,  1,  0],
     #   >>>         [ 2,  1,  0],
     #   >>>         [ 0, -1, -2]])
-    index = index * seq_mask
+    index = index * seq_mask.astype(index.dtype)
 
     #   >>> index
     #   >>> tensor([[2, 1, 0],