update docstring

05bc2588 · Hui Zhang · 6149daa2 · 05bc2588
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlespeech/s2t/models/u2/u2.py paddlespeech/s2t/models/u2/u2.py +1 -1

未找到文件。
--- 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)