Struct wyrm::IndexInputNode

pub struct IndexInputNode { /* fields omitted */ }

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

Methods

impl IndexInputNode

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

Create a new index input node.

Trait Implementations

impl Debug for IndexInputNode

fn fmt(&self, f: &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 clear(&self)

Reset the caches of this node and its parents.