Struct dfdx::tensor::OwnedTape

pub struct OwnedTape<E: Unit, D: DeviceStorage> { /* private fields */ }

Contains a Gradients and list of backward operations.

Trait Implementations

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

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

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

fn default() -> Self

Returns the “default value” for a type.

impl<E: Unit, D: DeviceStorage> Merge<NoneTape> for OwnedTape<E, D>

fn merge(self, _: NoneTape) -> Self

Merges T into self

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

fn merge(self, other: Self) -> Self

Merges T into self

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

const OWNS_TAPE: bool = true

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

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>