Helper functions to index tensors.
torch_index(self, indices)
self
(Tensor) Tensor that will be indexed.
indices
(List[Tensor]) List of indices. Indices are torch tensors with torch_long() dtype.
List[Tensor]
torch_long()