Skip to main content

expand

cubecl_std::tensor::index_offset_contiguous

Function expand 

Source 
pub fn expand<T: Scalar, N: Size>(
    scope: &mut Scope,
    tensor: <Tensor<Vector<T, N>> as CubeType>::ExpandType,
    offset_layout: <usize as CubeType>::ExpandType,
    rank: Option<usize>,
) -> <usize as CubeType>::ExpandType