Skip to content

GitLab

Commit 75514469 authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Update README.md

parent e87a2251
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ 75514469
...@@ -52,7 +52,7 @@ from torch_scatter import scatter_max ...@@ -52,7 +52,7 @@ from torch_scatter import scatter_max
src = torch.tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]]) 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]]) index = torch.tensor([[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]])
out, argmax = scatter_max(src, input) out, argmax = scatter_max(src, index)
``` ```
``` ```
......
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