reshape-for-vstack ( seq -- seq )

Vocabulary

tensors.private

Inputs

seq

an object

Outputs

seq

an object

Definition

USING: accessors arrays kernel math sequences tensors ;

IN: tensors.private

:: reshape-for-vstack ( seq -- seq )
    seq first dims 2 - :> vdim seq
    [ dup shape>> vdim cut product 1array append >>shape ] map!
    ;