numpy_map

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

def numpy_map(
        f: typ.Callable[
            [CarryType,
             typ.Sequence[NDArray],  # len=num_inputs
             ],
            typ.Tuple[
                CarryType,
                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[
    NDArray,  # shape[0]=map_length
    ...
]:  # len=num_outputs,

Parameters:

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

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

Return type:

Tuple[NDArray, Ellipsis]