Struct SegmentSumDescriptor 

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

Descriptor for a segment_sum op.

Fields

num_inputs: i32

Number of input rows (length of segment_ids).

embedding_dim: i32

Embedding / feature dim — second axis of input and output.

num_segments: i32

Total number of segments — first axis of output. Output is allocated for this many rows even when some segments are empty.

element: ElementKind

Value element type.

Trait Implementations

impl Clone for SegmentSumDescriptor

fn clone(&self) -> SegmentSumDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for SegmentSumDescriptor

impl Debug for SegmentSumDescriptor

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

Formats the value using the given formatter.