update docstring

3 years ago · 05bc258833
parent 6149daa221
commit 05bc258833
1 changed files with 1 additions and 1 deletions
--- 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)