torch_chunk

input

chunks

(int) dimension along which to split the tensor

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

 RStudio

torch_chunk function

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

Chunk

torch_chunk: Chunk

Description

Arguments

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

Description

Arguments

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

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