Struct GatherAddStage 

pub struct GatherAddStage {
    pub input_name: String,
    pub weight_key: String,
    pub axis: usize,
}

Add a gather-from-input embedding to the active hidden tensor.

Fields

input_name: String

weight_key: String

axis: usize

Implementations

impl GatherAddStage

pub fn new( input_name: impl Into<String>, weight_key: impl Into<String>, axis: usize, ) -> Self

Trait Implementations

impl Clone for GatherAddStage

fn clone(&self) -> GatherAddStage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GatherAddStage

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

Formats the value using the given formatter.