iteration-indices ( shaped -- seq )

Vocabulary

arrays.shaped

Inputs

shaped

an object

Outputs

seq

an object

Definition

USING: arrays kernel sequences ;

IN: arrays.shaped

: iteration-indices ( shaped -- seq )
    [ <iota> ] [
        cartesian-product concat
        [ dup first array? [ first2 suffix ] when ] map
    ] map-reduce ;