Struct PullbackPlan 

pub struct PullbackPlan<V: Differentiable> { /* private fields */ }

Compiled pullback execution plan.

Examples

let plan = chainrules::PullbackPlan::<MyType>::build(&loss).unwrap();

Implementations

impl<V: Differentiable> PullbackPlan<V>

pub fn build(_loss: &TrackedTensor<V>) -> AdResult<Self>

Builds a pullback plan from a loss value.

Examples

let plan = chainrules::PullbackPlan::build(&loss).unwrap();

pub fn execute(&self, _loss: &TrackedTensor<V>) -> AdResult<Gradients<V>>

Executes the pre-built pullback plan.

Examples

let grads = plan.execute(&loss).unwrap();

pub fn loss_node(&self) -> NodeId

Returns loss node ID for this plan.

Examples

use chainrules::{PullbackPlan, NodeId};
let _id_fn: fn(&PullbackPlan<f64>) -> NodeId = PullbackPlan::loss_node;

Trait Implementations

impl<V: Clone + Differentiable> Clone for PullbackPlan<V>

fn clone(&self) -> PullbackPlan<V>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<V: Debug + Differentiable> Debug for PullbackPlan<V>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.