Skip to content

GitLab

"...text-generation-inference.git" did not exist on "64347b05ff8b021c59c3d8417e67aafbe0a48c39"
Commit 328aaf2f authored by Maurits Diephuis's avatar Maurits Diephuis
Browse files

fixed a bracket placement

parent 1ffbee24
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_sparse/coalesce.py
View file @ 328aaf2f
...@@ -20,6 +20,6 @@ def coalesce(index, value, m, n, op="add"): ...@@ -20,6 +20,6 @@ def coalesce(index, value, m, n, op="add"):
""" """
storage = SparseStorage(row=index[0], col=index[1], value=value, storage = SparseStorage(row=index[0], col=index[1], value=value,
sparse_sizes=torch.Size([m, n], is_sorted=False)) sparse_sizes=torch.Size([m, n]), is_sorted=False)
storage = storage.coalesce(reduce=op) storage = storage.coalesce(reduce=op)
return torch.stack([storage.row(), storage.col()], dim=0), storage.value() return torch.stack([storage.row(), storage.col()], dim=0), storage.value()
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