fix (#4071)

4 months ago · e154b8f926
parent e59a3c1496
commit e154b8f926
1 changed files with 2 additions and 0 deletions
--- a/paddlespeech/t2s/modules/tacotron2/attentions.py
+++ b/paddlespeech/t2s/modules/tacotron2/attentions.py
@ -173,6 +173,8 @@ class AttLoc(nn.Layer):
            att_prev = 1.0 - make_pad_mask(enc_hs_len)
            att_prev = att_prev / enc_hs_len.unsqueeze(-1).astype(
                att_prev.dtype)
+        else:
+            att_prev.unsqueeze(-1)

        # att_prev: (utt, frame) -> (utt, 1, 1, frame)
        # -> (utt, att_conv_chans, 1, frame)