mindspore.Tensor.tensor_split

Tensor.tensor_split(indices_or_sections, axis=0)

For details, please refer to mindspore.ops.tensor_split().

Warning

Non-backward-compatible change after version 2.9.0: parameter name axis will be renamed to dim.