Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "0c18d02cc9c4803ed15d46ffc7521b91135cb300"
Commit 08dda1ad authored by rusty1s's avatar rusty1s
Browse files

coverage

parent 7d5034ba
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
torch_scatter/utils/gen.py
View file @ 08dda1ad
...@@ -20,7 +20,8 @@ def gen(src, index, dim=-1, out=None, dim_size=None, fill_value=0): ...@@ -20,7 +20,8 @@ def gen(src, index, dim=-1, out=None, dim_size=None, fill_value=0):
index_size[dim] = src.size(dim) index_size[dim] = src.size(dim)
if index.numel() > 0: if index.numel() > 0:
index = index.view(index_size).expand_as(src) index = index.view(index_size).expand_as(src)
else: # PyTorch has a bug when view is used on zero-element tensors. else: # pragma: no cover
# PyTorch has a bug when view is used on zero-element tensors.
index = src.new_empty(index_size, dtype=torch.long) index = src.new_empty(index_size, dtype=torch.long)
# Broadcasting capabilties: Expand dimensions to match. # Broadcasting capabilties: Expand dimensions to match.
......
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