Trait reverse::Gradient

pub trait Gradient<T, S> {
    fn wrt(&self, v: T) -> S;
}

Calculate gradients with respect to particular variables.

Required methods

fn wrt(&self, v: T) -> S

Calculate the gradient with respect to variable(s) v.

Implementations on Foreign Types

impl<'a> Gradient<&'_ Var<'a>, f64> for Vec<f64>

Calculate the gradient with respect to variable v.

fn wrt(&self, v: &Var<'_>) -> f64

impl<'a> Gradient<&'_ Vec<Var<'a>, Global>, Vec<f64, Global>> for Vec<f64>

Calculate the gradient with respect to all variables in v. Returns a vector, where the items in the vector are the gradients with respect to the variable in the original list v, in the same order.

fn wrt(&self, v: &Vec<Var<'a>>) -> Vec<f64>

impl<'a> Gradient<&'_ [Var<'a>], Vec<f64, Global>> for Vec<f64>

Calculate the gradient with respect to all variables in v. Returns a vector, where the items in the vector are the gradients with respect to the variable in the original list v, in the same order.

fn wrt(&self, v: &[Var<'a>]) -> Vec<f64>

impl<'a, const N: usize> Gradient<[Var<'a>; N], Vec<f64, Global>> for Vec<f64>

Calculate the gradient with respect to all variables in v. Returns a vector, where the items in the vector are the gradients with respect to the variable in the original list v, in the same order.

fn wrt(&self, v: [Var<'a>; N]) -> Vec<f64>

impl<'a, const N: usize> Gradient<&'_ [Var<'a>; N], Vec<f64, Global>> for Vec<f64>

Calculate the gradient with respect to all variables in v. Returns a vector, where the items in the vector are the gradients with respect to the variable in the original list v, in the same order.

fn wrt(&self, v: &[Var<'a>; N]) -> Vec<f64>

Implementors