评价此页

torch.Tensor.tensor_split#

Tensor.tensor_split(indices_or_sections, dim=0) List of Tensors#

参见 torch.tensor_split()