torch.Tensor.chunk

Tensor.chunk(chunks, dim=0) → List of Tensors

請參閱 torch.chunk()