Skip to main content

expand

cubecl_std::tensor::index_offset_contiguous_fastdivmod

Function expand 

Source 
pub fn expand(
    scope: &Scope,
    offset: <usize as CubeType>::ExpandType,
    shape: &<Sequence<FastDivmod<usize>> as CubeType>::ExpandType,
    stride: &<Sequence<usize> as CubeType>::ExpandType,
    vector_size: VectorSize,
) -> <usize as CubeType>::ExpandType
Expand description

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