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

上级 01f12622
......@@ -27,10 +27,6 @@ class GpuConnectionistTemporalClassification(gof.COp):
----------
compute_grad
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',)
......@@ -170,9 +166,8 @@ def gpu_ctc(activations, labels, input_lengths):
Returns
-------
1-D tensor
Cost of each example in the minibatch. Tensor is of shape
(time index, minibatch index, probabilities).
1-D array
Cost of each example in the minibatch.
"""
return GpuConnectionistTemporalClassification()(activations, labels,
input_lengths)
......
......@@ -27,10 +27,6 @@ class ConnectionistTemporalClassification(gof.COp, gof.OpenMPOp):
----------
compute_grad
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',)
......@@ -158,9 +154,8 @@ def ctc(activations, labels, input_lengths):
Returns
-------
1-D tensor
Cost of each example in the minibatch. Tensor is of shape
(time index, minibatch index, probabilities).
1-D array
Cost of each example in the minibatch.
"""
return ConnectionistTemporalClassification()(activations, labels, input_lengths)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论