map-diagonal ( shaped quot -- shaped )


Vocabulary
arrays.shaped

Inputs
shapedan object
quotan object


Outputs
shapedan object


Definition
USING: kernel sequences ;

IN: arrays.shaped

: map-diagonal ( shaped quot -- shaped )
[ check-2d ] dip [ [ first2 first2 = ] ] dip [ when ] curry
compose map-shaped-index ; inline