Struct burn::backend::autodiff::AutodiffBridge

pub struct AutodiffBridge<Bridge> { /* private fields */ }

Enable autodiff on a backend bridge.

Trait Implementations

impl<B, C, Bridge> BackendBridge<Autodiff<B, C>> for AutodiffBridge<Bridge>

where B: Backend, C: CheckpointStrategy, Bridge: BackendBridge<B> + 'static,

type Target = Autodiff<<Bridge as BackendBridge<B>>::Target, C>

The target backend

fn into_target( tensor: <Autodiff<B, C> as Backend>::FloatTensorPrimitive, _device: Option<<<AutodiffBridge<Bridge> as BackendBridge<Autodiff<B, C>>>::Target as Backend>::Device>, ) -> <<AutodiffBridge<Bridge> as BackendBridge<Autodiff<B, C>>>::Target as Backend>::FloatTensorPrimitive

Transfer the tensor to the target backend.

fn from_target( tensor: <<AutodiffBridge<Bridge> as BackendBridge<Autodiff<B, C>>>::Target as Backend>::FloatTensorPrimitive, _device: Option<<Autodiff<B, C> as Backend>::Device>, ) -> <Autodiff<B, C> as Backend>::FloatTensorPrimitive

Transfer the tensor from the target backend.

impl<Bridge> Debug for AutodiffBridge<Bridge>

where Bridge: Debug,

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

Formats the value using the given formatter.