Splits the tensor into chunks. Each chunk is a view of the original tensor.
torch_split(self, split_size, dim = 1L)
(Tensor) tensor to split.
(int) size of a single chunk or list of sizes for each chunk
(int) dimension along which to split the tensor.
split_size is an integer type, then
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 is a list, then
tensor will be split
length(split_size) chunks with sizes in
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.