Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "9dc1600b05ecf60fab556599b4c0bc6c32837449"
Commit af375340 authored by rusty1s's avatar rusty1s
Browse files

fixed div grad

parent 61eb4d03
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_scatter/functions/div.py
View file @ af375340
......@@ -12,7 +12,7 @@ class ScatterDiv(Scatter): # pragma: no cover
def backward_step(self, *data):
grad, output, index, input = data
return (grad / output.data).gather(self.dim, index.data) * input.data
return (output.data / grad).gather(self.dim, index.data) * input.data
def scatter_div_(output, index, input, dim=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