# Description

To resolve all environment indices, the previous implementation used
`self._ALL_INDICES`. However, this is not broadcastable when tensors are
indexed with two lists, for example:

```python
import torch

x = torch.zeros(100, 25)
x[[0, 2, 3, 4], [1, 2, 3]]
```

This MR fixes this bug (introduced in #248).

## Type of change

- Bug fix (non-breaking change which fixes an issue)

## Checklist

- [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./orbit.sh --format`
- [ ] I have made corresponding changes to the documentation
- [x] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have updated the changelog and the corresponding version in the
extension's `config/extension.toml` file
- [x] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there