t-concat ( seq -- tensor )
Tensors » Manipulating Tensors

Prev:

vstack ( seq -- tensor )

Vocabulary
tensors

Inputs

seq	a sequence

Outputs

tensor

a tensor

Word description
Joins the sequences in seq along the first axis.

Errors
Throws a shape-mismatch-error if the sequences in seq do not have the same shape along all but the first axis.

Definition

USING: accessors arrays kernel sequences tensors.private ;

IN: tensors

: t-concat ( seq -- tensor )
    check-concat-shape [
        [ [ shape>> first ] map-sum 1array ]
        [ first shape>> rest ] bi append
    ] [ [ vec>> ] map concat ] bi <tensor> ;