Function expand

pub fn expand<A: CubeType + CubeIndex<ExpandElementTyped<u32>>>(
    context: &mut CubeContext,
    array: ExpandElementTyped<A>,
    index: ExpandElementTyped<u32>,
) -> ExpandElementTyped<A::Output>
where
    A::Output: CubeType + Sized,