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: u32,
    dim_start: u32,
    dim_end: u32,
    unroll: bool,
) -> u32
Expand description

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