torch.Tensor.vsplit

Tensor.vsplit(split_size_or_sections) → List of Tensors

參見 torch.vsplit()