We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d847087 commit b5c8953Copy full SHA for b5c8953
README.md
@@ -70,7 +70,7 @@ from torch_scatter import scatter_max
70
src = torch.tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]])
71
index = torch.tensor([[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]])
72
73
-out, argmax = scatter_max(src, index)
+out, argmax = scatter_max(src, index, fill_value=0)
74
```
75
76
0 commit comments