From 00febff734bdc538fc8463c2ecf22b9c136b53f4 Mon Sep 17 00:00:00 2001 From: qingen Date: Tue, 19 Apr 2022 20:33:43 +0800 Subject: [PATCH] [vec][loss] update docstring, test=doc fix #1717 --- paddlespeech/vector/modules/loss.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/paddlespeech/vector/modules/loss.py b/paddlespeech/vector/modules/loss.py index 015c0dfe..af38dd01 100644 --- a/paddlespeech/vector/modules/loss.py +++ b/paddlespeech/vector/modules/loss.py @@ -132,6 +132,9 @@ class NCELoss(nn.Layer): def forward(self, output, target): """Forward inference + + Args: + output (tensor): the model output, which is the input of loss function """ output = paddle.reshape(output, [-1, self.N]) B = output.shape[0]