Trait dfdx::tensor::Tape

pub trait Tape<E, D: Storage<E>>: Default + Merge<Self> + Merge<NoneTape> {
    const OWNS_TAPE: bool;

    // Required method
    fn add_backward_op<F>(&mut self, operation: F)
       where F: 'static + FnOnce(&mut Gradients<E, D>) -> Result<(), D::Err>;
}

Something that can track backward operations.

Required Associated Constants

const OWNS_TAPE: bool

Whether this object is currently tracking gradients. This is known at compile time.

Required Methods

fn add_backward_op<F>(&mut self, operation: F)where F: 'static + FnOnce(&mut Gradients<E, D>) -> Result<(), D::Err>,

Implementors

impl<E, D: Storage<E>> Tape<E, D> for NoneTape

const OWNS_TAPE: bool = false

impl<E, D: Storage<E>> Tape<E, D> for OwnedTape<E, D>

const OWNS_TAPE: bool = true