From 05bc25883333d80a7ee1a5ec1314a1b81f57a81c Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Wed, 3 Aug 2022 09:17:23 +0000
Subject: [PATCH] update docstring

---
 paddlespeech/s2t/models/u2/u2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlespeech/s2t/models/u2/u2.py b/paddlespeech/s2t/models/u2/u2.py
index ca83ca170..e4c667e00 100644
--- a/paddlespeech/s2t/models/u2/u2.py
+++ b/paddlespeech/s2t/models/u2/u2.py
@@ -654,7 +654,7 @@ class U2BaseModel(ASRInterface, nn.Layer):
         Args:
             xs (paddle.Tensor): encoder output, (B, T, D)
         Returns:
-            paddle.Tensor: activation before ctc
+            paddle.Tensor: activation before ctc. (B, Tmax, odim)
         """
         return self.ctc.log_softmax(xs)