Skip to content

GitLab

Commit 4c8a4155 authored by rusty1s's avatar rusty1s
Browse files

fix default value for matmul

parent 2eff4072
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
torch_sparse/matmul.py
View file @ 4c8a4155
...@@ -109,9 +109,9 @@ def matmul(src: SparseTensor, other: Union[torch.Tensor, SparseTensor], ...@@ -109,9 +109,9 @@ def matmul(src: SparseTensor, other: Union[torch.Tensor, SparseTensor],
raise ValueError raise ValueError
SparseTensor.spmm = lambda self, other, reduce=None: spmm(self, other, reduce) SparseTensor.spmm = lambda self, other, reduce="sum": spmm(self, other, reduce)
SparseTensor.spspmm = lambda self, other, reduce=None: spspmm( SparseTensor.spspmm = lambda self, other, reduce="sum": spspmm(
self, other, reduce) self, other, reduce)
SparseTensor.matmul = lambda self, other, reduce=None: matmul( SparseTensor.matmul = lambda self, other, reduce="sum": matmul(
self, other, reduce) self, other, reduce)
SparseTensor.__matmul__ = lambda self, other: matmul(self, other, 'sum') SparseTensor.__matmul__ = lambda self, other: matmul(self, other, 'sum')
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment