Skip to main content

expand

ruprim::reduce::launch::reduce_kernel_virtual

Function expand 

Source 
pub fn expand<In: Numeric, InSize: Size, Out: Numeric, OutSize: Size, Acc: Numeric>(
    scope: &mut Scope,
    input: <VirtualTensor<In, InSize> as RudaType>::ExpandType,
    output: <VirtualTensor<Out, OutSize, ReadWrite> as RudaType>::ExpandType,
    reduce_axis: <usize as RudaType>::ExpandType,
    out_vec_axis: <usize as RudaType>::ExpandType,
    blueprint: ReduceBlueprint,
    config: ReduceOperationConfig,
) -> <() as RudaType>::ExpandType