broadcast-strides ( shape rank -- strides )

Vocabulary

arrays.shaped.private

Inputs

shape	an object
rank	an object

Outputs

strides

an object

Definition

USING: kernel sequences ;

IN: arrays.shaped.private

:: broadcast-strides ( shape rank -- strides )
    shape rank 1 pad-head dup row-major-strides
    [ swap 1 = [ drop 0 ] when ] 2map ;