torch.chunk
-
torch.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 bychunks
.
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.chunk.html