Skip to main content

Module reduce

Module reduce 

Source
Expand description

ReductionRequest<T> — dtype-generic tensor reduction.

Wraps cutensorCreateReduction + cutensorReduce. Output c holds the reduced result (shape determined by mode_cmode_a). op_reduce is one of CUTENSOR_OP_ADD, CUTENSOR_OP_MAX, CUTENSOR_OP_MIN, CUTENSOR_OP_MUL.

Structs§

ReductionRequest
Dtype-generic reduction request.

Functions§

build_reduction_key_raw