ragged_map

t3toolbox.backend.common.ragged_map(f, xs)

def ragged_map(
        f: typ.Callable[
            [
                typ.Sequence[NDArray],  # len=num_inputs
             ],
            typ.Tuple[
                typ.Sequence[NDArray],  # len=num_outputs
            ],
        ],
        xs: typ.Sequence[
            typ.Union[
                typ.Sequence[NDArray],  # len=map_length
                NDArray,  # shape[0]=map_length
            ]
        ],  # len=num_inputs
) -> typ.Tuple[
    typ.Tuple[NDArray, ...],  # len=map_length
    ...
]:  # len=num_outputs

Parameters:

• f (Callable[[Sequence[NDArray]], Tuple[Sequence[NDArray]]])

• xs (Sequence[Union[Sequence[NDArray], NDArray]])

Return type:

Tuple[Tuple[NDArray, Ellipsis], Ellipsis]