Skip to main content

expand

cubecl_std::tensor::index_offset_with_layout

Function expand 

Source 
pub fn expand<T: Scalar, N1: Size, L: Scalar, N2: Size>(
    scope: &mut Scope,
    tensor: <Tensor<Vector<T, N1>> as CubeType>::ExpandType,
    layout: <Tensor<Vector<L, N2>> as CubeType>::ExpandType,
    offset_layout: <usize as CubeType>::ExpandType,
    dim_start: <usize as CubeType>::ExpandType,
    dim_end: <usize as CubeType>::ExpandType,
    unroll: bool,
) -> <usize as CubeType>::ExpandType