Trait autograd::op::Op

pub trait Op {
    fn name(&self) -> &str;
    fn compute(&self, ctx: OpComputeContext) -> ComputeResult;
    fn grad(
        &self, 
        gy: &Tensor, 
        xs: &[&Tensor], 
        y: &Tensor
    ) -> Vec<Option<Tensor>>;
}

Operation trait. Tensor wraps trait-object of this.

Required Methods

fn name(&self) -> &str

Name of this op

fn compute(&self, ctx: OpComputeContext) -> ComputeResult

Actually runs this op.

Important traits for Vec<u8>

impl Write for Vec<u8>

fn grad(&self, gy: &Tensor, xs: &[&Tensor], y: &Tensor) -> Vec<Option<Tensor>>

Returns symbolic gradients for input nodes by use of output gradient etc.

Arguments

• gy - Symbolic representation of the gradient of compute's return value
• xs - Symbolic representation of compute::xs
• y - Symbolic representation of compute's return value

NOTE: The number of return values must match xs.len().

Implementors