reduce_trace_axes

Function reduce_trace_axes 

Source
pub fn reduce_trace_axes<T, Op>(
    src: &StridedView<'_, T, Op>,
    trace_axes: &[usize],
) -> Result<StridedArray<T>>
where T: Copy + Send + Sync + Add<Output = T> + Zero, Op: ElementOp<T>,
Expand description

Reduce all trace axes from a view by summing them out.

trace_axes are indices into the original view’s dimensions, given in ascending order. Each axis is reduced (summed) in sequence from back to front so that axis indices remain valid after each reduction.

Returns a new StridedArray with the trace axes removed.