Trait burn_tensor::backend::Gradients

pub trait Gradients<B: ADBackend>: Send + Sync {
    fn empty() -> Self;
    fn get<const D: usize>(
        &self,
        id: &str
    ) -> Option<&Tensor<B::InnerBackend, D>>;
    fn register<const D: usize>(
        &mut self,
        id: String,
        value: Tensor<B::InnerBackend, D>
    );
}

Required Methods

fn empty() -> Self

fn get<const D: usize>(&self, id: &str) -> Option<&Tensor<B::InnerBackend, D>>

fn register<const D: usize>(
    &mut self,
    id: String,
    value: Tensor<B::InnerBackend, D>
)

Implementors