Fixed "perform" function of StructuredDotCSR

上级 04662229
......@@ -876,7 +876,7 @@ class StructuredDotCSR(gof.Op):
return r
def perform(self, node, (a_val, a_ind, a_ptr, a_ncols, b), (out,)):
a = sparse.csc_matrix((a_val, a_ind, a_ptr),
a = sparse.csr_matrix((a_val, a_ind, a_ptr),
(a_ncols, b.shape[0]),
copy = False)
out[0] = a.dot(b)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论