ArraySplit ========== Split an array into sub-arrays. **Abstract Signature:** ``ArraySplit(ary: Tensor, indices_or_sections: int | Sequence[int], axis: int = 0)`` .. raw:: html
torch.tensor_split
jax.numpy.array_split
numpy.array_split