hstack-unsafe ( tseq -- tensor )

Vocabulary

tensors.private

Inputs

tseq

an object

Outputs

tensor

an object

Definition

USING: accessors kernel math sequences sequences.extras tensors
;

IN: tensors.private

:: hstack-unsafe ( tseq -- tensor )
    tseq final-hstack-shape (tensor) tseq hstack-guide dup
    length :> repeat :> guide dup vec>> [
        :> i drop i repeat /mod guide nth dup first tseq nth
        [ [ second ] [ third ] bi -rot * + ] dip nth
    ] map-index! drop ;