spekk.transformations.common.get_args_for_index

spekk.transformations.common.get_args_for_index(args: Sequence, in_axes: Sequence[int | None], i: int) → Sequence