torch.Tensor.dsplit

Tensor.dsplit(split_size_or_sections) → Tensor 列表

参见 torch.dsplit()