final-vstack-shape ( seq -- shape )

Vocabulary

tensors.private

Inputs

seq

an object

Outputs

shape

an object

Definition

USING: accessors kernel math sequences tensors ;

IN: tensors.private

:: final-vstack-shape ( seq -- shape )
    seq first dims 2 - :> vdim seq
    [ shape>> vdim swap nth ] map-sum
    seq first shape>> clone :> new-shape vdim new-shape set-nth
    new-shape ;