提交 613c3547 authored 作者: David Warde-Farley's avatar David Warde-Farley

Fix csm_grad infer_shape test: use right CSM class.

上级 fbb29e66
...@@ -119,7 +119,8 @@ class SparseInferShapeTester(unittest.TestCase): ...@@ -119,7 +119,8 @@ class SparseInferShapeTester(unittest.TestCase):
y = tensor.ivector() y = tensor.ivector()
z = tensor.ivector() z = tensor.ivector()
s = tensor.ivector() s = tensor.ivector()
spm = sp.csr_matrix(random_lil((200, 400), config.floatX, 5)) call = getattr(sp, sparsetype + '_matrix')
spm = call(random_lil((300, 400), config.floatX, 5))
out = tensor.grad(dense_from_sparse( out = tensor.grad(dense_from_sparse(
CSM(sparsetype)(x, y, z, s) CSM(sparsetype)(x, y, z, s)
).sum(), x) ).sum(), x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论