pub fn expand<C: CubePrimitive + Default + IntoRuntime>(
scope: &mut Scope,
tensor: <Tensor<C> as CubeType>::ExpandType,
index: <usize as CubeType>::ExpandType,
unroll_factor: usize,
kernel_name: String,
) -> <C as CubeType>::ExpandType