Struct wyrm::IndexInputNode

pub struct IndexInputNode {
    pub value: RefCell<SmallVec<[usize; 4]>>,
}

An input node for integer indices into ParameterNodes, used for implementing indexable embedding layers.

Fields

value: RefCell<SmallVec<[usize; 4]>>

Methods

impl IndexInputNode

fn new(value: &[usize]) -> Variable<Self>

Create a new index input node.

Trait Implementations

impl Debug for IndexInputNode

fn fmt(&self, __arg_0: &mut Formatter) -> Result

Formats the value using the given formatter.

impl Node for IndexInputNode

type Value = SmallVec<[usize; 4]>

Type of the node's value.

type InputGradient = Arr

Type of the input gradient the node receives during backpropagation.

fn forward(&self)

Perform the forward step. Should recursively call the forward methods of its ancestors.

fn backward(&self, _: &Ref<Self::InputGradient>)

Perform the backward step. Should recursively call the backward methods of its ancestors.

fn value(&self) -> Bor<Self::Value>

Return the value of the node.

fn needs_gradient(&self) -> bool

If the node needs to be used in the backward step.

fn zero_gradient(&self)