How to optimize tensor indexing

4

For the moment I use this solution : Batched index_select but I’m also curious if there is a better alternative.

2 Likes