identity-matrix ( n -- shaped )

Vocabulary

arrays.shaped

Inputs

n

an object

Outputs

shaped

an object

Definition

USING: arrays kernel sequences ;

IN: arrays.shaped

: identity-matrix ( n -- shaped )
    dup 2array zeros [ second first2 = 1 0 ? ] map-shaped-index
    ;