(Batched) advanced indexing for PyTorch. - View it on GitHub