提交 ae29f845 authored 作者: Frederic's avatar Frederic

don't use numpy.zeros_like dtype parameter as it was added in numpy 1.6.0

上级 b64abd85
...@@ -854,7 +854,7 @@ class CSMGrad(gof.op.Op): ...@@ -854,7 +854,7 @@ class CSMGrad(gof.op.Op):
sp_dim = x_shape[0] sp_dim = x_shape[0]
g_row = numpy.zeros(sp_dim, dtype=g_data.dtype) g_row = numpy.zeros(sp_dim, dtype=g_data.dtype)
gout_data = numpy.zeros_like(x_data, dtype=node.outputs[0].dtype) gout_data = numpy.zeros(x_data.shape, dtype=node.outputs[0].dtype)
for i in range(len(x_indptr) - 1): for i in range(len(x_indptr) - 1):
for j_ptr in range(g_indptr[i], g_indptr[i + 1]): for j_ptr in range(g_indptr[i], g_indptr[i + 1]):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论