Skip to content

GitLab

Unverified Commit fe38ab58 authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

Update denseToSparse.py re issue 220

parent 6752a3d5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
sparseconvnet/denseToSparse.py
View file @ fe38ab58
...@@ -55,7 +55,7 @@ class DenseToSparseFunction(Function): ...@@ -55,7 +55,7 @@ class DenseToSparseFunction(Function):
s = torch.LongTensor(nz.stride()).view(1, dimension + 1) s = torch.LongTensor(nz.stride()).view(1, dimension + 1)
nz = nz.nonzero() nz = nz.nonzero()
s = s.type_as(nz) s = s.type_as(nz)
aa = aa.view(-1, input.size(1)) aa = aa.reshape(-1, input.size(1))
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)
......
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