[Fix] type promotion

1 year ago · 88daa60159
parent 5e03da403b
commit 88daa60159
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],