Skip to main content

expand

cubecl_std::tensor::index_offset_contiguous

Function expand 

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

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