pub struct ComputationTape { /* private fields */ }
Expand description
Computational tape for recording operations
Implementations§
Source§impl ComputationTape
impl ComputationTape
Sourcepub fn backward(&self, gradients: &mut Vec<f64>) -> Result<(), OptimizeError>
pub fn backward(&self, gradients: &mut Vec<f64>) -> Result<(), OptimizeError>
Perform backpropagation to compute gradients
Sourcepub fn forward(&self, input_values: &[f64]) -> Result<Vec<f64>, OptimizeError>
pub fn forward(&self, input_values: &[f64]) -> Result<Vec<f64>, OptimizeError>
Forward pass to compute all variable values
Sourcepub fn add_constant(&mut self, value: f64) -> usize
pub fn add_constant(&mut self, value: f64) -> usize
Add a constant to the tape
Sourcepub fn add_unary_op(
&mut self,
op_type: UnaryOpType,
input: usize,
input_values: &[f64],
) -> usize
pub fn add_unary_op( &mut self, op_type: UnaryOpType, input: usize, input_values: &[f64], ) -> usize
Add a unary operation with automatic partial derivative computation
Sourcepub fn add_binary_op(
&mut self,
op_type: BinaryOpType,
left: usize,
right: usize,
input_values: &[f64],
) -> usize
pub fn add_binary_op( &mut self, op_type: BinaryOpType, left: usize, right: usize, input_values: &[f64], ) -> usize
Add a binary operation with automatic partial derivative computation
Sourcepub fn forward_ad(
&self,
input_values: &[f64],
seed_derivatives: &[f64],
) -> Result<(Vec<f64>, Vec<f64>), OptimizeError>
pub fn forward_ad( &self, input_values: &[f64], seed_derivatives: &[f64], ) -> Result<(Vec<f64>, Vec<f64>), OptimizeError>
Forward-mode AD: compute function value and derivatives simultaneously
Trait Implementations§
Source§impl Debug for ComputationTape
impl Debug for ComputationTape
Auto Trait Implementations§
impl Freeze for ComputationTape
impl RefUnwindSafe for ComputationTape
impl Send for ComputationTape
impl Sync for ComputationTape
impl Unpin for ComputationTape
impl UnwindSafe for ComputationTape
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self
into a Left
variant of Either<Self, Self>
if into_left
is true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self
into a Left
variant of Either<Self, Self>
if into_left(&self)
returns true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self
from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self
is actually part of its subset T
(and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset
but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self
to the equivalent element of its superset.Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self
from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self
is actually part of its subset T
(and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset
but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self
to the equivalent element of its superset.