Struct SparsemaxDescriptor 

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

Descriptor for a Sparsemax forward op.

Fields

input_shape: [i32; N]

Tensor shape (input and output share it).

softmax_axis: u8

Axis along which to apply sparsemax. Must be in [0, N). Extent along this axis must be <= SPARSEMAX_MAX_EXTENT.

element: ElementKind

Element type.

Trait Implementations

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

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

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

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

Formats the value using the given formatter.