torch_index: Index torch tensors

View source: R/gen-namespace.R

torch_indexR 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.


torch documentation built on May 29, 2024, 9:54 a.m.