torch_index function

Index torch tensors

Helper functions to index tensors.

torch_index(self, indices)

Arguments

• self: (Tensor) Tensor that will be indexed.
• indices: (List[Tensor]) List of indices. Indices are torch tensors with torch_long() dtype.