Trait dfdx::tensor::Tape

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

    // Required methods
    fn add_backward_op<F>(&mut self, operation: F)
       where F: 'static + FnOnce(&mut Gradients<E, D>) -> Result<(), D::Err>;
    fn try_alloc_grad<S: Shape>(
        &mut self,
        t: &Tensor<S, 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>,

fn try_alloc_grad<S: Shape>( &mut self, t: &Tensor<S, E, D> ) -> Result<(), D::Err>

Implementors

impl<E: Unit, D: DeviceStorage> Tape<E, D> for NoneTape

const OWNS_TAPE: bool = false

impl<E: Unit, D: DeviceStorage> Tape<E, D> for OwnedTape<E, D>

const OWNS_TAPE: bool = true