check-reshape ( shape1 shape2 -- shape1 shape2 )

Vocabulary

tensors.private

Inputs

shape1	an object
shape2	an object

Outputs

shape1	an object
shape2	an object

Definition

USING: kernel sequences tensors ;

IN: tensors.private

: check-reshape ( shape1 shape2 -- shape1 shape2 )
    2dup [ product ] bi@ = [ shape-mismatch-error ] unless ;