提交 75ec37f3 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix structureddot csr grad test

上级 0ada7dff
......@@ -524,9 +524,9 @@ class test_structureddot(unittest.TestCase):
verify_grad_sparse(structured_dot, [spmat, mat], structured=True)
def buildgraph_T(spmat, mat):
rval = structured_dot(mat.T, spmat.T)
return structured_dot(mat.T, spmat.T)
verify_grad_sparse(buildgraph_T, [spmat.data, mat], structured=True)
verify_grad_sparse(buildgraph_T, [spmat, mat], structured=True)
def test_infer_shape_csr_csc_grad(self):
for sparsetype in ('csr', 'csc'):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论