From 84c004172d604ecb966e1a905b81d6e1db2fc2e0 Mon Sep 17 00:00:00 2001
From: dathudeptrai <nguyenquananhminh@gmail.com>
Date: Thu, 9 Jul 2020 23:18:17 +0700
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20Fix=20#106?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tensorflow_tts/models/tacotron2.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tensorflow_tts/models/tacotron2.py b/tensorflow_tts/models/tacotron2.py
index 1cb8a0fc..bed69d1a 100644
--- a/tensorflow_tts/models/tacotron2.py
+++ b/tensorflow_tts/models/tacotron2.py
@@ -907,6 +907,10 @@ def inference(self, input_ids, input_lengths, speaker_ids):
 
         mel_outputs = decoder_output + residual_projection
 
+        alignment_history = tf.transpose(
+            final_decoder_state.alignment_history.stack(), [1, 2, 0]
+        )
+
         return decoder_output, mel_outputs, stop_token_prediction, alignment_history
 
     @tf.function(