unstack

penzai.core.named_axes.unstack(array: NamedArrayBase, axis_name: AxisName) → Sequence[NamedArray | NamedArrayView]

Splits a named array across a given named axis.

Parameters:

• array – The array to split.

• axis_name – Name of the axis along which to unstack.

Returns:

A sequence of slices across the given axis.