Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/fairseq.git" did not exist on "8afb77612ce35fd9a3457c3faaaf1c61195ea8df"
Commit ae35b8a5 authored by rusty1s's avatar rusty1s
Browse files

remove dim_size

parent cdd1cb4d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
torch_sparse/storage.py
View file @ ae35b8a5
...@@ -269,8 +269,7 @@ class SparseStorage(object): ...@@ -269,8 +269,7 @@ class SparseStorage(object):
value = self.value value = self.value
if self.has_value(): if self.has_value():
idx = mask.cumsum(0) - 1 idx = mask.cumsum(0) - 1
dim_size = idx[-1].item() + 1 value = segment_csr(idx, value, reduce=reduce)
value = segment_csr(idx, value, dim_size=dim_size, reduce=reduce)
value = value[0] if isinstance(value, tuple) else value value = value[0] if isinstance(value, tuple) else value
return self.__class__(index, value, self.sparse_size(), is_sorted=True) return self.__class__(index, value, self.sparse_size(), is_sorted=True)
......
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