typos

README.md

@@ -13,13 +13,18 @@ pip install torch_scatter
 ```py
 from torch_scatter import scatter_max
 
-input = torch.Tensor([[2, 0, 1, 4, 3], [0,2, 1, 3, 4]])
-torch.LongTensor([[4, 5, 2, 3], [0, 0, 2, 2, 1]])
+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, arg = scatter_max_(index, input, dim=1)
-```
+max, argmax = scatter_max_(index, input, dim=1)
 
-## Features
+print(max)
+ 0  0  4  3  2  0
+ 2  4  3  0  0  0
+[torch.FloatTensor of size 2x6]
 
-* CPU and Cuda implementation
-* Backward implementation
+print(argmax)
+-1 -1  3  4  0  1
+ 1  4  3 -1 -1 -1
+[torch.LongTensor of size 2x6]
+```