Remove Returns header from CTC classes and fix return type doc of ctc functions

上级 01f12622
...@@ -27,10 +27,6 @@ class GpuConnectionistTemporalClassification(gof.COp): ...@@ -27,10 +27,6 @@ class GpuConnectionistTemporalClassification(gof.COp):
---------- ----------
compute_grad compute_grad
If set to True, enables the computation of gradients of the CTC loss function. If set to True, enables the computation of gradients of the CTC loss function.
Returns
-------
GPU Op
An instance of the GPU CTC loss computation Op
""" """
__props__ = ('compute_grad',) __props__ = ('compute_grad',)
...@@ -170,9 +166,8 @@ def gpu_ctc(activations, labels, input_lengths): ...@@ -170,9 +166,8 @@ def gpu_ctc(activations, labels, input_lengths):
Returns Returns
------- -------
1-D tensor 1-D array
Cost of each example in the minibatch. Tensor is of shape Cost of each example in the minibatch.
(time index, minibatch index, probabilities).
""" """
return GpuConnectionistTemporalClassification()(activations, labels, return GpuConnectionistTemporalClassification()(activations, labels,
input_lengths) input_lengths)
......
...@@ -27,10 +27,6 @@ class ConnectionistTemporalClassification(gof.COp, gof.OpenMPOp): ...@@ -27,10 +27,6 @@ class ConnectionistTemporalClassification(gof.COp, gof.OpenMPOp):
---------- ----------
compute_grad compute_grad
If set to True, enables the computation of gradients of the CTC loss function. If set to True, enables the computation of gradients of the CTC loss function.
Returns
-------
Op
An instance of the CTC loss computation Op
""" """
__props__ = ('compute_grad',) __props__ = ('compute_grad',)
...@@ -158,9 +154,8 @@ def ctc(activations, labels, input_lengths): ...@@ -158,9 +154,8 @@ def ctc(activations, labels, input_lengths):
Returns Returns
------- -------
1-D tensor 1-D array
Cost of each example in the minibatch. Tensor is of shape Cost of each example in the minibatch.
(time index, minibatch index, probabilities).
""" """
return ConnectionistTemporalClassification()(activations, labels, input_lengths) return ConnectionistTemporalClassification()(activations, labels, input_lengths)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论