index_offset_with_layout

cubecl_std::tensor

Function index_offset_with_layout 

Source 
pub fn index_offset_with_layout<N: CubePrimitive, L: CubePrimitive>(
    tensor: &Tensor<Line<N>>,
    layout: &Tensor<Line<L>>,
    offset_layout: usize,
    dim_start: usize,
    dim_end: usize,
    unroll: bool,
) -> usize
Expand description

Returns the offset of the tensor corresponding to the layout tensor.