Struct ReduceBackwardDescriptor 

pub struct ReduceBackwardDescriptor<const N: usize> {
    pub kind: ReduceKind,
    pub input_shape: [i32; N],
    pub reduce_axis: u8,
    pub element: ElementKind,
    pub correction: i32,
}

Descriptor for a single-axis reduction backward.

Fields

kind: ReduceKind

Which forward reduction this is the backward of.

input_shape: [i32; N]

Shape of the forward input (= shape of dx).

reduce_axis: u8

Axis that was reduced. Must satisfy 0 <= reduce_axis < N.

element: ElementKind

Element type.

correction: i32

Bessel correction for Var / Std BW only. 1 = sample variance (PyTorch default), 0 = population variance. Ignored by other reductions.

Implementations

impl<const N: usize> ReduceBackwardDescriptor<N>

pub fn dy_shape(&self) -> [i32; N]

Compute the keepdim dy shape (input shape with reduce_axis = 1).

Trait Implementations

impl<const N: usize> Clone for ReduceBackwardDescriptor<N>

fn clone(&self) -> ReduceBackwardDescriptor<N>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<const N: usize> Copy for ReduceBackwardDescriptor<N>

impl<const N: usize> Debug for ReduceBackwardDescriptor<N>

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

Formats the value using the given formatter.