torch_split

self

(int) size of a single chunk or list of sizes for each chunk

split_size

(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_split function

Splits the tensor into chunks. Each chunk is a view of the original tensor.<pre>If `split_size_or_sections` is an integer type, then `tensor` will
be split into equally sized chunks (if possible). Last chunk will be smaller if
the tensor size along the given dimension `dim` is not divisible by
`split_size`.If `split_size_or_sections` is a list, then `tensor` will be split
into `len(split_size_or_sections)` chunks with sizes in `dim` according
to `split_size_or_sections`.
</pre>

torch_split: Split

Description

Usage

Arguments

TEST