torch_index {torch}R Documentation

Index torch tensors

Description

Helper functions to index tensors.

Usage

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.

[Package torch version 0.13.0 Index]