unstack — Python array API standard 2023.12 documentation (original) (raw)

unstack(x: array, /, *, axis: int = 0) → Tuple[array, ...]¶

Splits an array into a sequence of arrays along the given axis.

Parameters:

• x (array) – input array.
• axis (int) – axis along which the array will be split. A valid axis must be on the interval [-N, N), where N is the rank (number of dimensions) of x. If provided an axis outside of the required interval, the function must raise an exception. Default: 0.

Returns:

out (Tuple[array, …]) – tuple of slices along the given dimension. All the arrays have the same shape.

Notes

New in version 2023.12.