oa::ml::GradientTape

Thread-affine reverse-mode recording scope. Constructing a tape selects it for operations on the current thread. GradientTape::backward closes the recording scope and records the corresponding backward operations into the originating engine’s eager batch.

StRust public API

Declaration

pub struct GradientTape { /* private fields */ }
Public Rust declaration.

Public Methods

pub fn backward(&self, root: &Matrix) -> Result<()>
pub fn close(&self)
fn drop(&mut self)

Associated Functions

pub fn new() -> Self
fn default() -> Self

Examples

Public Method Documentation

pub fn backward(&self, root: &Matrix) -> Result<()>
No public source comment is attached to this declaration.
pub fn close( &self )
No public source comment is attached to this declaration.
fn drop( &mut self )
No public source comment is attached to this declaration.
pub fn new() -> Self
No public source comment is attached to this declaration.
fn default() -> Self
No public source comment is attached to this declaration.