Struct ScaleAdd

pub struct ScaleAdd<RHS, LHS>
where
    RHS: Node,
    LHS: Node,
{
    pub rhs: RefCell<RHS>,
    pub lhs: RefCell<LHS>,
    pub output: RefCell<Value<NDArray<f64>>>,
    pub gradient: RefCell<Value<NDArray<f64>>>,
}

Fields

rhs: RefCell<RHS>

lhs: RefCell<LHS>

output: RefCell<Value<NDArray<f64>>>

gradient: RefCell<Value<NDArray<f64>>>

Implementations

impl<RHS, LHS> ScaleAdd<RHS, LHS>

where RHS: Node, LHS: Node,

pub fn new(rhs: RHS, lhs: LHS) -> Self

Create new instance of elememtwise add operation

pub fn rhs(&self) -> RefMut<'_, dyn Node>

Retrieve right hand side value of elementwise add operation

pub fn lhs(&self) -> RefMut<'_, dyn Node>

Retrieve left hand side value of elementwise add operation

Trait Implementations

impl<LHS, RHS> Node for ScaleAdd<RHS, LHS>

where RHS: Node, LHS: Node,

fn forward(&mut self)

Perform forward pass of elementwise add operation

fn backward(&mut self, upstream_gradient: NDArray<f64>)

Perform backward pass of elementwise add operation

fn value(&self) -> NDArray<f64>

Get output value of elementwise add operation

fn grad(&self) -> NDArray<f64>

Get gradient of elementwise add operation

fn set_grad(&mut self, upstream_gradient: NDArray<f64>)

Set gradient of elementwise add operation