Index torch tensors — torch

Helper functions to index tensors.

Usage

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.