From a1867c20c341e7305c5442d3617a8ab1f900346e Mon Sep 17 00:00:00 2001
From: TianYuan <white-sky@qq.com>
Date: Thu, 13 Jan 2022 14:21:29 +0800
Subject: [PATCH] fix slice bug of speedyspeech expand, test=tts (#1337)

---
 paddlespeech/t2s/models/speedyspeech/speedyspeech.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddlespeech/t2s/models/speedyspeech/speedyspeech.py b/paddlespeech/t2s/models/speedyspeech/speedyspeech.py
index 3e64e670..cc9e2066 100644
--- a/paddlespeech/t2s/models/speedyspeech/speedyspeech.py
+++ b/paddlespeech/t2s/models/speedyspeech/speedyspeech.py
@@ -30,7 +30,9 @@ def expand(encodings: paddle.Tensor, durations: paddle.Tensor) -> paddle.Tensor:
         k = 0
         for j in range(t_enc):
             d = durations[i, j]
-            M[i, k:k + d, j] = 1
+            # If the d == 0, slice action is meaningless and not supported
+            if d >= 1:
+                M[0, k:k + d, j] = 1
             k += d
     encodings = paddle.matmul(M, encodings)
     return encodings