Skip to content

GitLab

Commit 0e482107 authored by rusty1s's avatar rusty1s
Browse files

typo

parent 454556cd
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
README.md
View file @ 0e482107
...@@ -17,7 +17,9 @@ input = torch.Tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]]) ...@@ -17,7 +17,9 @@ input = torch.Tensor([[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]])
index = torch.LongTensor([[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]]) index = torch.LongTensor([[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]])
max, argmax = scatter_max_(index, input, dim=1) max, argmax = scatter_max_(index, input, dim=1)
```
```
print(max) print(max)
0 0 4 3 2 0 0 0 4 3 2 0
2 4 3 0 0 0 2 4 3 0 0 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