DSplit

Split array into multiple sub-arrays along the 3rd axis (depth).

Abstract Signature:

DSplit(ary: Tensor, indices_or_sections: int)

PyTorch

API: torch.dsplit
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.dsplit
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.dsplit
Strategy: Direct Mapping
Official Docs ↗

Keras

API: —
Strategy: Custom / Partial

TensorFlow

API: tf.experimental.numpy.dsplit
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: —
Strategy: Custom / Partial