basic_ragged_unstack

t3toolbox.backend.stacking.basic_ragged_unstack(x, first_leaf_num_nonstacking_axes)

def basic_ragged_unstack(
        x:                               typ.Tuple[
            typ.Tuple[NDArray, ...],     # stacked ragged array tuple (leading axes are the stack)
            ...,
        ],
        first_leaf_num_nonstacking_axes: int,  # ndim of x[0][0] excluding stack axes -> infers stack rank
):                                       # array-like tree of ragged array tuples

Unstack stacked ragged array tuple into array-like tree

Parameters:

• x (t3toolbox.backend.common.typ.Tuple[t3toolbox.backend.common.typ.Tuple[NDArray, Ellipsis], Ellipsis])

• first_leaf_num_nonstacking_axes (int)