Skip to content

Commit 481fb99

Browse files
committed
fix
1 parent 64a8e2c commit 481fb99

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

torch_sparse/view.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,10 @@ def _view(src: SparseTensor, n: int, layout: str = 'csr') -> SparseTensor:
88
row, col, value = src.coo()
99
sparse_sizes = src.storage.sparse_sizes()
1010

11-
if sparse_sizes[0] * sparse_sizes[1] % n == 0:
11+
if sparse_sizes[0] * sparse_sizes[1] % n != 0:
1212
raise RuntimeError(
13-
f"shape '[-1, {n}]' is invalid for input of size {sparse_sizes[0] * sparse_sizes[1]}")
13+
f"shape '[-1, {n}]' is invalid for input of size "
14+
f"{sparse_sizes[0] * sparse_sizes[1]}")
1415

1516
assert layout == 'csr' or layout == 'csc'
1617

0 commit comments

Comments
 (0)