About the index of tensor

I remember there is an indexing library that supports wrapping of tensor in pytorch and tensorflow and extending its indexing method, I wonder if anyone remembers what the name of this library is?