Add outputs list to construct GPU CTC Op's outputs

上级 2715fdfb
...@@ -120,14 +120,16 @@ class GpuConnectionistTemporalClassification(gof.COp): ...@@ -120,14 +120,16 @@ class GpuConnectionistTemporalClassification(gof.COp):
costs = GpuArrayType(dtype='float32', costs = GpuArrayType(dtype='float32',
broadcastable=(False,), broadcastable=(False,),
context_name=context)() context_name=context)()
outputs = [costs]
if self.compute_grad: if self.compute_grad:
gradients = GpuArrayType(dtype='float32', gradients = GpuArrayType(dtype='float32',
broadcastable=(False, False, False,), broadcastable=(False, False, False,),
context_name=context)() context_name=context)()
outputs += [gradients]
return theano.Apply(self, inputs=[t_activations, t_labels, t_input_lengths], return theano.Apply(self, inputs=[t_activations, t_labels, t_input_lengths],
outputs=[costs, gradients]) outputs=outputs)
def L_op(self, inputs, outputs, output_grads): def L_op(self, inputs, outputs, output_grads):
if not ctc_enabled: if not ctc_enabled:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论