pub fn expand<N: CubePrimitive>(
    scope: &mut Scope,
    tensor: <Tensor<Line<N>> as CubeType>::ExpandType,
    offset_layout: <u32 as CubeType>::ExpandType,
    shape: <Sequence<FastDivmod> as CubeType>::ExpandType,
    stride: <Sequence<u32> as CubeType>::ExpandType,
) -> <u32 as CubeType>::ExpandType