torch_index

Helper functions to index tensors.

Provides functionality to define and train neural networks similar to
'PyTorch' by Paszke et al (2019) <arXiv:1912.01703> but written entirely in R
using the 'libtorch' library. Also supports low-level tensor operations and
'GPU' acceleration.

Daniel Falbel

torch

Tensors and Neural Networks with 'GPU' Acceleration

Javier Luraschi

Dmitriy Selivanov

Athos Damiani

Christophe Regouby

Krzysztof Joachimiak

Hamada S. Badr

 RStudio

torch_index function

<dl><dt>self</dt>
<dd>(Tensor) Tensor that will be indexed.</dd>
<dt>indices</dt>
<dd>(<code>List[Tensor]</code>) List of indices. Indices are torch tensors with
<code>torch_long()</code> dtype.</dd></dl>

Arguments

Index torch tensors — torch_index

<dl>

<dt>self</dt>
<dd>(Tensor) Tensor that will be indexed.</dd>


<dt>indices</dt>
<dd>(<code>List[Tensor]</code>) List of indices. Indices are torch tensors with
<code>torch_long()</code> dtype.</dd>

</dl>

torch_index: Index torch tensors

Description

Usage

Arguments