expand

pub fn expand(
    scope: &mut Scope,
    offset: <u32 as CubeType>::ExpandType,
    shape: <Sequence<FastDivmod> as CubeType>::ExpandType,
    stride: <Sequence<u32> as CubeType>::ExpandType,
    line_size: u32,
) -> <u32 as CubeType>::ExpandType