Skip to content

GitLab

Unverified Commit 0fc7dc2d authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #22 from ekagra-ranjan/patch-3 

Fix bug in `coalesced` arg of transpose
parents 7c55386b 82fee5a2
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_sparse/transpose.py
View file @ 0fc7dc2d
...@@ -23,6 +23,6 @@ def transpose(index, value, m, n, coalesced=True): ...@@ -23,6 +23,6 @@ def transpose(index, value, m, n, coalesced=True):
row, col = index row, col = index
index = torch.stack([col, row], dim=0) index = torch.stack([col, row], dim=0)
if coalesce: if coalesced:
index, value = coalesce(index, value, n, m) index, value = coalesce(index, value, n, m)
return index, value return index, 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