torch.Tensor.hsplit

Tensor.hsplit(split_size_or_sections) → 张量列表

参见 torch.hsplit()