torch_chunk

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_chunk function

<dl><dt>self</dt>
<dd>(Tensor) the tensor to split</dd>
<dt>chunks</dt>
<dd>(int) number of chunks to return</dd>
<dt>dim</dt>
<dd>(int) dimension along which to split the tensor</dd></dl>

Arguments

Splits a tensor into a specific number of chunks. Each chunk is a view of
the input tensor.
Last chunk will be smaller if the tensor size along the given dimension
<code>dim</code> is not divisible by <code>chunks</code>.

chunk(input, chunks, dim=0) -&gt; List of Tensors 

Chunk — torch_chunk

<dl>

<dt>self</dt>
<dd>(Tensor) the tensor to split</dd>


<dt>chunks</dt>
<dd>(int) number of chunks to return</dd>


<dt>dim</dt>
<dd>(int) dimension along which to split the tensor</dd>

</dl>

Splits a tensor into a specific number of chunks. Each chunk is a view of
the input tensor.
Last chunk will be smaller if the tensor size along the given dimension
<code>dim</code> is not divisible by <code>chunks</code>.

The Learning Leader's Guide to AI Literacy

torch_chunk: Chunk

Description

Usage

Arguments

chunk(input, chunks, dim=0) -> List of Tensors

The Learning Leader's Guide to AI Literacy

Description

Usage

Arguments

chunk(input, chunks, dim=0) -&gt; List of Tensors

chunk(input, chunks, dim=0) -> List of Tensors