Struct UnsortedSegmentProdDescriptor 

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

Descriptor for an unsorted_segment_prod 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 UnsortedSegmentProdDescriptor

fn clone(&self) -> UnsortedSegmentProdDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for UnsortedSegmentProdDescriptor

impl Debug for UnsortedSegmentProdDescriptor

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

Formats the value using the given formatter.