Replace pygpu_empty with pygpu_zeros in gradients allocation inside CTC C wrapper

上级 7b762a4a
...@@ -198,7 +198,7 @@ int APPLY_SPECIFIC(ctc_cost_gpu)(PyGpuArrayObject * in_activations, ...@@ -198,7 +198,7 @@ int APPLY_SPECIFIC(ctc_cost_gpu)(PyGpuArrayObject * in_activations,
Py_XDECREF( *out_gradients ); Py_XDECREF( *out_gradients );
const size_t * activation_dims = PyGpuArray_DIMS( in_activations ); const size_t * activation_dims = PyGpuArray_DIMS( in_activations );
*out_gradients = pygpu_empty( 3, activation_dims, GA_FLOAT, GA_C_ORDER, *out_gradients = pygpu_zeros( 3, activation_dims, GA_FLOAT, GA_C_ORDER,
ctx, Py_None ); ctx, Py_None );
if ( NULL == *out_gradients ) if ( NULL == *out_gradients )
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论