Trait LinearizedOp 

pub trait LinearizedOp<V>:
    Send
    + Sync
    + 'static
where
    V: Differentiable + Send + Sync + 'static,
{
    // Required methods
    fn jvp(
        &self,
        input_tangents: &[Option<<V as Differentiable>::Tangent>],
    ) -> Result<Vec<Option<<V as Differentiable>::Tangent>>, AutodiffError>;
    fn vjp(
        &self,
        output_cotangents: &[Option<<V as Differentiable>::Tangent>],
        input_grad_mask: &[bool],
    ) -> Result<Vec<Option<<V as Differentiable>::Tangent>>, AutodiffError>;
}

Required Methods

fn jvp( &self, input_tangents: &[Option<<V as Differentiable>::Tangent>], ) -> Result<Vec<Option<<V as Differentiable>::Tangent>>, AutodiffError>

fn vjp( &self, output_cotangents: &[Option<<V as Differentiable>::Tangent>], input_grad_mask: &[bool], ) -> Result<Vec<Option<<V as Differentiable>::Tangent>>, AutodiffError>

Implementors