Trait burn::module::ADModule

pub trait ADModule: Module<Backend = Self::ADBackend> + Send + Sync + Debug + Display {
    type ADBackend: ADBackend;
    type InnerModule: Module<Backend = <Self::ADBackend as ADBackend>::InnerBackend>;

    fn inner(&self) -> Self::InnerModule;
}

Module with auto-differentiation backend.

Required Associated Types

type ADBackend: ADBackend

type InnerModule: Module<Backend = <Self::ADBackend as ADBackend>::InnerBackend>

Required Methods

fn inner(&self) -> Self::InnerModule

Get the same module, but on the inner backend without auto-differentiation.

Implementors

impl<B: Backend> ADModule for LayerNorm<B>where
    B: ADBackend,

type ADBackend = B

type InnerModule = LayerNorm<<B as ADBackend>::InnerBackend>

impl<B: Backend> ADModule for Linear<B>where
    B: ADBackend,

type ADBackend = B

type InnerModule = Linear<<B as ADBackend>::InnerBackend>