"vscode:/vscode.git/clone" did not exist on "35d39735da9013a953959ae04dfd12db25200cb4"
Commit b5c89536 authored by rusty1s's avatar rusty1s
Browse files

changed fill_value

parent d8470875
......@@ -70,7 +70,7 @@ from torch_scatter import scatter_max
src = torch.tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]])
index = torch.tensor([[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]])
out, argmax = scatter_max(src, index)
out, argmax = scatter_max(src, index, fill_value=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