oneflow.Tensor.split

Tensor.split(split_size_or_sections=None, dim=0)

See oneflow.split()