torch.Tensor.vsplit

Tensor.vsplit(split_size_or_sections) → List of Tensors

参见 torch.vsplit()