Struct SoftmaxBackwardDescriptor 

pub struct SoftmaxBackwardDescriptor<const N: usize> {
    pub kind: SoftmaxKind,
    pub input_shape: [i32; N],
    pub softmax_axis: u8,
    pub element: ElementKind,
}

Descriptor for a softmax-family BW op.

Fields

kind: SoftmaxKind

Which softmax variant this is the BW of.

input_shape: [i32; N]

Tensor shape (shared by dy / y / dx).

softmax_axis: u8

Forward softmax axis.

element: ElementKind

Element type.

Trait Implementations

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

fn clone(&self) -> SoftmaxBackwardDescriptor<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 SoftmaxBackwardDescriptor<N>

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

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

Formats the value using the given formatter.