Struct Reduction 

pub struct Reduction;

A reduction op: D[mD] = reduce(A[mA]) with user-chosen reduce op.

Implementations

impl Reduction

pub unsafe fn new<'h>( handle: &'h Handle, a: &TensorDescriptor<'h>, modes_a: &[i32], c: &TensorDescriptor<'h>, modes_c: &[i32], d: &TensorDescriptor<'h>, modes_d: &[i32], op_reduce: BinaryOp, compute_desc: *const c_void, ) -> Result<OperationDescriptor<'h>>

Build a reduction. modes_d is a subset of modes_a — all modes in a that do NOT appear in d are reduced. op_reduce is ADD for sum, MUL for product, MAX/MIN for min-max.

Safety

compute_desc must be null or valid.

Trait Implementations

impl Debug for Reduction

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.