We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 76d04d6 + 668dd20 commit bccaa10Copy full SHA for bccaa10
torch_sparse/coalesce.py
@@ -20,6 +20,6 @@ def coalesce(index, value, m, n, op="add"):
20
"""
21
22
storage = SparseStorage(row=index[0], col=index[1], value=value,
23
- sparse_sizes=torch.Size([m, n], is_sorted=False))
+ sparse_sizes=torch.Size([m, n]), is_sorted=False)
24
storage = storage.coalesce(reduce=op)
25
return torch.stack([storage.row(), storage.col()], dim=0), storage.value()
0 commit comments