Fix docstring on the return type of CTC Op

上级 02072e34
......@@ -39,7 +39,7 @@ class GpuConnectionistTemporalClassification(gof.COp):
Returns
-------
3D tensor
1-D tensor
Cost of each example in the minibatch. Tensor is of shape
(time index, minibatch index, probabilities).
"""
......
......@@ -39,7 +39,7 @@ class ConnectionistTemporalClassification(gof.COp, gof.OpenMPOp):
Returns
-------
3D tensor
1-D tensor
Cost of each example in the minibatch. Tensor is of shape
(time index, minibatch index, probabilities).
"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论