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

PyTorch

API: torch.tensor_split
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.array_split
Strategy: Direct Mapping

NumPy

API: numpy.array_split
Strategy: Direct Mapping