matmul-batch-shape ( left right -- shape )

Vocabulary

arrays.shaped.private

Inputs

left	an object
right	an object

Outputs

shape

an object

Definition

USING: arrays.shaped kernel math.order sequences ;

IN: arrays.shaped.private

:: matmul-batch-shape ( left right -- shape )
    left length right length max :> rank left rank 1 pad-head
    :> a right rank 1 pad-head :> b a b
    [ compatible-dimensions? ] 2all?
    [ left right shape-mismatch ] unless a b
    [ over 1 = [ nip ] [ drop ] if ] 2map ;