Update denseToSparse.py re issue 220

emepetres · Jan 26, 2022 · fe38ab5 · fe38ab5
1 parent 6752a3d
commit fe38ab5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparseconvnet/denseToSparse.py b/sparseconvnet/denseToSparse.py
@@ -55,7 +55,7 @@ def forward(
         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)