diff --git a/paddlespeech/t2s/modules/tacotron2/attentions.py b/paddlespeech/t2s/modules/tacotron2/attentions.py index 72d1b1375..7a9e833bb 100644 --- a/paddlespeech/t2s/modules/tacotron2/attentions.py +++ b/paddlespeech/t2s/modules/tacotron2/attentions.py @@ -174,7 +174,7 @@ class AttLoc(nn.Layer): att_prev = att_prev / enc_hs_len.unsqueeze(-1).astype( att_prev.dtype) else: - att_prev.unsqueeze(-1) + att_prev.unsqueeze(-1) # only for static graph # att_prev: (utt, frame) -> (utt, 1, 1, frame) # -> (utt, att_conv_chans, 1, frame)