Struct UnsortedSegmentSumDescriptor 

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

Descriptor for an unsorted_segment_sum op.

Fields

num_inputs: i32

Number of input rows.

embedding_dim: i32

Embedding / feature dim.

num_segments: i32

Total number of segments (output rows). Out-of-range seg-ids in the input are silently dropped.

element: ElementKind

Value element type.

Trait Implementations

impl Clone for UnsortedSegmentSumDescriptor

fn clone(&self) -> UnsortedSegmentSumDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for UnsortedSegmentSumDescriptor

impl Debug for UnsortedSegmentSumDescriptor

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

Formats the value using the given formatter.