Struct SoftmaxDescriptor 

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

Descriptor for a softmax-family op.

Fields

kind: SoftmaxKind

Which softmax variant.

input_shape: [i32; N]

Tensor shape — input and output share it.

softmax_axis: u8

Axis along which to compute softmax. Must be in [0, N).

element: ElementKind

Element type.

Trait Implementations

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

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

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

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

Formats the value using the given formatter.