Function index_offset_with_layout 

pub fn index_offset_with_layout<T: Scalar, N1: Size, L: Scalar, N2: Size>(
    tensor: &Tensor<Vector<T, N1>>,
    layout: &Tensor<Vector<L, N2>>,
    offset_layout: usize,
    dim_start: usize,
    dim_end: usize,
    unroll: bool,
) -> usize

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