cubecl_std::tensor::index_offset_with_layout

Function expand

Source 
pub fn expand<N: CubePrimitive, L: CubePrimitive>(
    scope: &mut Scope,
    tensor: <Tensor<Line<N>> as CubeType>::ExpandType,
    layout: <Tensor<Line<L>> as CubeType>::ExpandType,
    offset_layout: <u32 as CubeType>::ExpandType,
    dim_start: <u32 as CubeType>::ExpandType,
    dim_end: <u32 as CubeType>::ExpandType,
    unroll: bool,
) -> <u32 as CubeType>::ExpandType