Update storage.py

d24a4533 · Mario Geiger · GitHub · 9586b8be · d24a4533
Unverified Commit d24a4533 authored May 14, 2020 by Mario Geiger Committed by GitHub May 14, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torch_sparse/storage.py torch_sparse/storage.py +1 -1

No files found.
--- a/torch_sparse/storage.py
+++ b/torch_sparse/storage.py
@@ -277,7 +277,7 @@ class SparseStorage(object):

        idx = self.sparse_size(1) * self.row() + self.col()

-        row = idx / num_cols
+        row = idx // num_cols
        col = idx % num_cols

        return SparseStorage(row=row, rowptr=None, col=col, value=self._value,