提交 42cc94fe authored 作者: Frederic Bastien's avatar Frederic Bastien

Small crash fix to new sandbox sparse code.

上级 5dfeaf73
...@@ -135,8 +135,8 @@ class SquareDiagonal(Op): ...@@ -135,8 +135,8 @@ class SquareDiagonal(Op):
""" """
def make_node(self, diag): def make_node(self, diag):
return gof.Apply(self, [diag], return gof.Apply(self, [diag],
[sparse.Sparse(dtype = diag.dtype, [sparse.SparseType(dtype = diag.dtype,
format = 'csc').make_result()]) format = 'csc')()])
def perform(self, node, (diag,), (z,)): def perform(self, node, (diag,), (z,)):
N, = diag.shape N, = diag.shape
indptr = range(N+1) indptr = range(N+1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论