torch.Tensor.vsplit

Tensor.vsplit(split_size_or_sections) → Tensor 列表

参见 torch.vsplit()