Update denseToSparse.py re issue 220

fe38ab58 · Ben Graham · GitHub · 6752a3d5 · fe38ab58
Unverified Commit fe38ab58 authored Jan 26, 2022 by Ben Graham Committed by GitHub Jan 26, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

sparseconvnet/denseToSparse.py sparseconvnet/denseToSparse.py +1 -1

No files found.
--- a/sparseconvnet/denseToSparse.py
+++ b/sparseconvnet/denseToSparse.py
@@ -55,7 +55,7 @@ class DenseToSparseFunction(Function):
        s = torch.LongTensor(nz.stride()).view(1, dimension + 1)
        nz = nz.nonzero()
        s = s.type_as(nz)
-        aa = aa.view(-1, input.size(1))
+        aa = aa.reshape(-1, input.size(1))
        ctx.aas2 = aa.size()
        r = (nz * s.expand_as(nz)).sum(1).view(-1)
        output_features = aa.index_select(0, r)