check-concat-shape ( seq -- seq )

Vocabulary

tensors.private

Inputs

seq

an object

Outputs

seq

an object

Definition

USING: accessors kernel sequences tensors ;

IN: tensors.private

:: check-concat-shape ( seq -- seq )
    seq first { } >tensor dup :> empty-tensor like
    shape>> dup :> first-shape rest :> rest-shape seq [
        empty-tensor like dup shape>> rest rest-shape =
        [ shape>> first-shape swap shape-mismatch-error ] unless
    ] map ;