torch: torch_chunk – R documentation

Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!

torch_chunk

Chunk

Description

Chunk

Usage

torch_chunk(self, chunks, dim = 1L)

Arguments

`self`	(Tensor) the tensor to split
`chunks`	(int) number of chunks to return
`dim`	(int) dimension along which to split the tensor

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

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 dim is not divisible by chunks.

torch

Tensors and Neural Networks with 'GPU' Acceleration

v0.3.0

MIT + file LICENSE

Authors

Daniel Falbel [aut, cre, cph], Javier Luraschi [aut], Dmitriy Selivanov [ctb], Athos Damiani [ctb], Christophe Regouby [ctb], Krzysztof Joachimiak [ctb], RStudio [cph]

Initial release