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