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

typo

parent 454556cd
......@@ -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]])
max, argmax = scatter_max_(index, input, dim=1)
```
```
print(max)
0 0 4 3 2 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