Struct Value
pub struct Value<V>where
V: Differentiable,{ /* private fields */ }Expand description
Public value handle for reverse-mode AD.
Value exposes a torch-like surface while keeping graph ownership hidden.
Internally it carries either a leaf gradient sink or an edge into a reverse
graph.
Implementations§
§impl<V> Value<V>where
V: Differentiable + 'static,
impl<V> Value<V>where
V: Differentiable + 'static,
pub fn requires_grad(&self) -> bool
pub fn requires_grad(&self) -> bool
Return whether this value participates in reverse-mode AD.
pub fn with_requires_grad(self, enabled: bool) -> Value<V>
pub fn with_requires_grad(self, enabled: bool) -> Value<V>
Return a new value handle with gradient tracking enabled or disabled.
pub fn grad(
&self,
) -> Result<Option<<V as Differentiable>::Tangent>, AutodiffError>where
<V as Differentiable>::Tangent: Clone,
pub fn grad(
&self,
) -> Result<Option<<V as Differentiable>::Tangent>, AutodiffError>where
<V as Differentiable>::Tangent: Clone,
Read the accumulated leaf gradient, if available.
pub fn backward(&self) -> Result<(), AutodiffError>where
<V as Differentiable>::Tangent: Clone,
pub fn backward(&self) -> Result<(), AutodiffError>where
<V as Differentiable>::Tangent: Clone,
Run reverse-mode backward with the default scalar cotangent seed.
pub fn backward_with_seed(
&self,
seed: <V as Differentiable>::Tangent,
) -> Result<(), AutodiffError>where
<V as Differentiable>::Tangent: Clone,
pub fn backward_with_seed(
&self,
seed: <V as Differentiable>::Tangent,
) -> Result<(), AutodiffError>where
<V as Differentiable>::Tangent: Clone,
Run reverse-mode backward with an explicit cotangent seed.
pub fn grad_wrt_with_seed(
&self,
seed: <V as Differentiable>::Tangent,
wrt: &[&Value<V>],
) -> Result<Vec<Option<<V as Differentiable>::Tangent>>, AutodiffError>where
<V as Differentiable>::Tangent: Clone,
pub fn grad_wrt_with_seed(
&self,
seed: <V as Differentiable>::Tangent,
wrt: &[&Value<V>],
) -> Result<Vec<Option<<V as Differentiable>::Tangent>>, AutodiffError>where
<V as Differentiable>::Tangent: Clone,
Compute functional gradients with respect to the requested inputs.
Return whether self and other share any reachable reverse graph.
Auto Trait Implementations§
impl<V> !Freeze for Value<V>
impl<V> RefUnwindSafe for Value<V>where
V: RefUnwindSafe,
impl<V> Send for Value<V>
impl<V> Sync for Value<V>
impl<V> Unpin for Value<V>
impl<V> UnwindSafe for Value<V>where
V: RefUnwindSafe,
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
§impl<T> DistributionExt for Twhere
T: ?Sized,
impl<T> DistributionExt for Twhere
T: ?Sized,
fn rand<T>(&self, rng: &mut (impl Rng + ?Sized)) -> Twhere
Self: Distribution<T>,
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 more