[Fix] type promotion (#3817)

10 months ago · 6316514329
parent 658d19a73e
commit 6316514329
1 changed files with 1 additions and 1 deletions
--- 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],