clathe

### 🐛 Describe the bug I believe the function `group_argsort` returns the wrong values. If there is only one index, the behaviour should be equivalent to `torch.argsort` but this is...