Skip to content

GitLab

"tools/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "edb82b9f7ec575aebd2e8d9c704ca911d6197f9d"
Unverified Commit bb5de2f5 authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

Fix DenseToSparse

parent 3da6f2df
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
sparseconvnet/denseToSparse.py
View file @ bb5de2f5
...@@ -59,8 +59,7 @@ class DenseToSparseFunction(Function): ...@@ -59,8 +59,7 @@ class DenseToSparseFunction(Function):
ctx.aas2 = aa.size() ctx.aas2 = aa.size()
r = (nz * s.expand_as(nz)).sum(1).view(-1) r = (nz * s.expand_as(nz)).sum(1).view(-1)
output_features = aa.index_select(0, r) output_features = aa.index_select(0, r)
dim_fn(dimension, 'createMetadataForDenseToSparse')( output_metadata.createMetadataForDenseToSparse(
output_metadata,
output_spatial_size, output_spatial_size,
nz.cpu(), nz.cpu(),
input.size(0)) input.size(0))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment