Trait DifferentiableArray 

pub trait DifferentiableArray: ArrayProtocol {
    // Required methods
    fn gradient(
        &self,
        variables: &[Box<dyn DifferentiableArray>],
    ) -> Vec<Box<dyn DifferentiableArray>>;
    fn set_requiresgrad(&mut self, requiresgrad: bool);
    fn requiresgrad(&self) -> bool;
    fn grad(&self) -> Option<Box<dyn DifferentiableArray>>;
}

Trait for arrays that support automatic differentiation.

Required Methods

fn gradient( &self, variables: &[Box<dyn DifferentiableArray>], ) -> Vec<Box<dyn DifferentiableArray>>

Compute the gradient of this array with respect to some variables.

fn set_requiresgrad(&mut self, requiresgrad: bool)

Set whether to record operations for automatic differentiation.

fn requiresgrad(&self) -> bool

Check if this array requires gradient computation.

fn grad(&self) -> Option<Box<dyn DifferentiableArray>>

Get the gradient of this array.

Implementors