Fix input_lengths instatiation in make_node of CTC Op

上级 3721b260
...@@ -69,8 +69,7 @@ class ConnectionistTemporalClassification(gof.COp): ...@@ -69,8 +69,7 @@ class ConnectionistTemporalClassification(gof.COp):
t_activations = cpu_contiguous(t_activations) t_activations = cpu_contiguous(t_activations)
t_labels = T.as_tensor_variable(labels) t_labels = T.as_tensor_variable(labels)
t_input_lengths = T.cast(activations.shape[0], dtype="int32") * \ t_input_lengths = T.as_tensor_variable(input_lengths)
T.ones_like(activations[0, :, 0], dtype=np.int32)
# Return only the cost. Gradient will be returned by grad() # Return only the cost. Gradient will be returned by grad()
self.default_output = 0 self.default_output = 0
...@@ -92,7 +91,7 @@ class ConnectionistTemporalClassification(gof.COp): ...@@ -92,7 +91,7 @@ class ConnectionistTemporalClassification(gof.COp):
grad_undefined(self, 2, inputs[2])] grad_undefined(self, 2, inputs[2])]
def ctc(activations, labels, input_lengths=None): def ctc(activations, labels, input_lengths):
return ConnectionistTemporalClassification()(activations, labels, input_lengths) return ConnectionistTemporalClassification()(activations, labels, input_lengths)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论