Skip to main content

HasAutodiffModule

burn_core::module

Trait HasAutodiffModule 

Source 
pub trait HasAutodiffModule<B: AutodiffBackend> {
    type TrainModule: AutodiffModule<B, InnerModule = Self>;
}
Expand description

Helper trait to associate a module with its autodiff version.

Required Associated Types§

Source

type TrainModule: AutodiffModule<B, InnerModule = Self>

The module with auto-differentiation.

Dyn Compatibility§

This trait is dyn compatible.

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

Implementors§