Skip to contents

Chunk

Source: R/gen-namespace-docs.R, R/gen-namespace-examples.R, R/gen-namespace.R
torch_chunk.Rd

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.