Fix DenseToSparse

bb5de2f5 · Ben Graham · GitHub · 3da6f2df · bb5de2f5
Unverified Commit bb5de2f5 authored Jun 26, 2018 by Ben Graham Committed by GitHub Jun 26, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

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

No files found.
--- a/sparseconvnet/denseToSparse.py
+++ b/sparseconvnet/denseToSparse.py
@@ -59,8 +59,7 @@ class DenseToSparseFunction(Function):
        ctx.aas2 = aa.size()
        r = (nz * s.expand_as(nz)).sum(1).view(-1)
        output_features = aa.index_select(0, r)
-        dim_fn(dimension, 'createMetadataForDenseToSparse')(
-            output_metadata,
+        output_metadata.createMetadataForDenseToSparse(
            output_spatial_size,
            nz.cpu(),
            input.size(0))