Module dfdx::gradients

Expand description

Implementations of GradientTape and generic Nd array containers via Gradients.

Structs

Records gradient computations to execute later.

A generic container for keeping variable sized arrays associated with a UniqueId.

Contains nothing. When Tape::add_backward_op is called, this struct does nothing.

Contains a boxed GradientTape. When Tape::add_backward_op is called, this function passes the operation directly to GradientTape.

Holds UniqueId of tensors that were missing gradients during CanUpdateWithGradients::update(), and therefore are unused

Represents something that can be updated with GradientProvider.

Represents something that can return a gradient for a given key.

Something that can add a gradient operation to GradientTape.