Struct wyrm::InputNode

pub struct InputNode {
    pub value: RefCell<Arr>,
}

Input node for the graph.

Fields

value: RefCell<Arr>

Methods

impl InputNode

fn new(value: Arr) -> Variable<Self>

Create a new input node with a given value. This fixes the shape of the node in the graph.

Trait Implementations

impl Debug for InputNode

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

Formats the value using the given formatter.

impl Node for InputNode

type Value = Arr

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)