Struct wyrm::ParameterNode

pub struct ParameterNode { /* fields omitted */ }

Parameter node, holds the optimizable parameters of the model.

Methods

impl ParameterNode

pub fn shared(value: Arc<HogwildParameter>) -> Variable<Self>

Create a parameter node that shares its parameter values with other parameter nodes via the HogwildParameter object.

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

Create a new parameter node. The parameters held by this node cannot be shared and optimized in parallel.

Trait Implementations

impl Debug for ParameterNode

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

Formats the value using the given formatter.

impl Node for ParameterNode

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, gradient: &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.