Skip to main content

reduce

Function reduce 

Source
pub fn reduce<R: Runtime>(
    input: &RudaTensor<R>,
    input_modes: &[Mode],
    output_modes: &[Mode],
    reduction: ReductionOp,
) -> Result<RudaTensor<R>>
Expand description

Reduce modes absent from output_modes, in the specified output order.