Browse Source

!13398 complement the docs of CTCLoss operator.

From: @wangshuide2020
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @liangchenghui
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
debbfb5134
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ops/operations/nn_ops.py

+ 3
- 0
mindspore/ops/operations/nn_ops.py View File

@@ -7185,6 +7185,9 @@ class CTCLoss(PrimitiveWithInfer):
"""
Calculates the CTC (Connectionist Temporal Classification) loss and the gradient.

The CTC algorithm is proposed in `Connectionist Temporal Classification: Labeling Unsegmented Sequence Data with
Recurrent Neural Networks <http://www.cs.toronto.edu/~graves/icml_2006.pdf>`_.

Args:
preprocess_collapse_repeated (bool): If true, repeated labels will be collapsed prior to the CTC calculation.
Default: False.


Loading…
Cancel
Save