Struct UnsortedSegmentMaxDescriptor 

pub struct UnsortedSegmentMaxDescriptor {
    pub num_inputs: i32,
    pub embedding_dim: i32,
    pub num_segments: i32,
    pub element: ElementKind,
}

Descriptor for an unsorted_segment_max op.

Fields

num_inputs: i32

Number of input rows.

embedding_dim: i32

Embedding / feature dim.

num_segments: i32

Total number of segments.

element: ElementKind

Value element type.

Trait Implementations

impl Clone for UnsortedSegmentMaxDescriptor

fn clone(&self) -> UnsortedSegmentMaxDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for UnsortedSegmentMaxDescriptor

impl Debug for UnsortedSegmentMaxDescriptor

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

Formats the value using the given formatter.