未验证 提交 78738d6c 编写于 作者: W whs 提交者: GitHub

Fix comments of ctc_greedy_decoder. (#14679)

test=develop
上级 1b9753d1
...@@ -4251,7 +4251,14 @@ def ctc_greedy_decoder(input, blank, name=None): ...@@ -4251,7 +4251,14 @@ def ctc_greedy_decoder(input, blank, name=None):
input.lod = [[4, 4]] input.lod = [[4, 4]]
Then: Computation:
step1: Apply argmax to first input sequence which is input.data[0:4]. Then we get:
[[0], [2], [1], [0]]
step2: merge repeated tokens and remove blank which is 0. Then we get first output sequence:
[[2], [1]]
Finally:
output.data = [[2], output.data = [[2],
[1], [1],
...@@ -4259,6 +4266,7 @@ def ctc_greedy_decoder(input, blank, name=None): ...@@ -4259,6 +4266,7 @@ def ctc_greedy_decoder(input, blank, name=None):
output.lod = [[2, 1]] output.lod = [[2, 1]]
Args: Args:
input(Variable): (LoDTensor<float>), the probabilities of input(Variable): (LoDTensor<float>), the probabilities of
...@@ -4273,8 +4281,10 @@ def ctc_greedy_decoder(input, blank, name=None): ...@@ -4273,8 +4281,10 @@ def ctc_greedy_decoder(input, blank, name=None):
name (str): The name of this layer. It is optional. name (str): The name of this layer. It is optional.
Returns: Returns:
Variable: CTC greedy decode result. If all the sequences in result were Variable: CTC greedy decode result which is a 2-D tensor with shape [Lp, 1].
empty, the result LoDTensor will be [-1] with LoD [[]] and dims [1, 1]. 'Lp' is the sum if all output sequences' length. If all the sequences
in result were empty, the result LoDTensor will be [-1] with
LoD [[]] and dims [1, 1].
Examples: Examples:
.. code-block:: python .. code-block:: python
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册