output-shape ( sa0 sa1 -- shape )

Vocabulary

arrays.shaped

Inputs

sa0	a shaped-array
sa1	a shaped-array

Outputs

shape

a sequence

Word description

Returns the broadcast result shape, aligning trailing axes and treating missing leading axes as size one. Corresponding dimensions must be equal or one must be one. A zero dimension combined with one remains zero. Incompatible shapes raise shape-mismatch.

Definition

USING: arrays.shaped.private kernel sequences ;

IN: arrays.shaped

:: output-shape ( sa0 sa1 -- shape )
    sa0 sa1 broadcastable? [
        sa0 sa1 aligned-shapes
        [ over 1 = [ nip ] [ drop ] if ] 2map
    ] [ sa0 sa1 shape-mismatch ] if ;