Trait Backward 

pub trait Backward<B, const N: usize>:
    Send
    + Debug
    + Sized
    + 'static
where
    B: Backend,
{
    type State: Clone + Send + Debug + 'static;

    // Required method
    fn backward(
        self,
        ops: Ops<Self::State, N>,
        grads: &mut Gradients,
        checkpointer: &mut Checkpointer,
    );

    // Provided method
    fn prepare<C>(
        self,
        nodes: [NodeGuard; N],
    ) -> OpsPrep<Self, B, Self::State, C, N>
       where C: CheckpointStrategy { ... }
}

Trait for all operations.

Notes

Concrete types implementing this trait should not have any state. If a state is necessary during the backward pass, they should be declared with the associated type ‘State’.

Required Associated Types

type State: Clone + Send + Debug + 'static

Associated type to compute the backward pass.

Required Methods

fn backward( self, ops: Ops<Self::State, N>, grads: &mut Gradients, checkpointer: &mut Checkpointer, )

The backward pass.

Provided Methods

fn prepare<C>( self, nodes: [NodeGuard; N], ) -> OpsPrep<Self, B, Self::State, C, N>

where C: CheckpointStrategy,

Prepare the backward ops, retaining input references through registration.

Obtain guards from the input tensors before consuming their primitives.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors