more readable

上级 c6ef502e
...@@ -46,16 +46,11 @@ class sparse2dense(op): ...@@ -46,16 +46,11 @@ class sparse2dense(op):
def gen_outputs(self): return [core.NumpyR()] def gen_outputs(self): return [core.NumpyR()]
def impl(x): return numpy.asarray(x.todense()) def impl(x): return numpy.asarray(x.todense())
def grad(self, x, gz): def grad(self, x, gz):
if x.format is sparse.coo_matrix: if x.format is sparse.coo_matrix: return dense2coo(gz)
return dense2coo(gz) if x.format is sparse.csc_matrix: return dense2csc(gz)
if x.format is sparse.csc_matrix: if x.format is sparse.csr_matrix: return dense2csr(gz)
return dense2csc(gz) if x.format is sparse.dok_matrix: return dense2dok(gz)
if x.format is sparse.csr_matrix: if x.format is sparse.lil_matrix: return dense2lil(gz)
return dense2csr(gz)
if x.format is sparse.dok_matrix:
return dense2dok(gz)
if x.format is sparse.lil_matrix:
return dense2lil(gz)
# convert an ndarray to various sorts of sparse matrices. # convert an ndarray to various sorts of sparse matrices.
class _dense2sparse(op): class _dense2sparse(op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论