ArraySplit

Split an array into sub-arrays.

Abstract Signature:

ArraySplit(ary: Tensor, indices_or_sections: int | Sequence[int], axis: int = 0)

PyTorchJAX (Core)NumPy

PyTorch

API: torch.tensor_split

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.array_split

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.array_split

Strategy: Direct Mapping

Official Docs ↗