Skip to content

GitLab

Commit af673237 authored by rusty1s's avatar rusty1s
Browse files

cleaner

parent fdba2556
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
torch_scatter/functions/scatter.py
View file @ af673237
...@@ -36,8 +36,7 @@ class _Scatter(Function): ...@@ -36,8 +36,7 @@ class _Scatter(Function):
if len(grad) == 3: if len(grad) == 3:
return grad_output, None, grad_input return grad_output, None, grad_input
else: return grad_output, None, grad_input, None
return grad_output, None, grad_input, None
def scatter(name, dim, *data): def scatter(name, dim, *data):
......
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