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.
Splits the tensor into chunks. Each chunk is a view of the original tensor.
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`.