SubModel1

Trait SubModel1 

Source
pub trait SubModel1 {
    type Config;
    type Input;
    type Output;

    // Required methods
    fn build(vb: VarBuilder<'_>, config: Self::Config) -> Self;
    fn forward(&self, input: &Self::Input) -> Self::Output;
}
Expand description

Neural network model not owing its VarMap internally.

Required Associated Types§

Source

type Config

Configuration from which SubModel1 is constructed.

Source

type Input

Input of the SubModel1.

Source

type Output

Output of the SubModel1.

Required Methods§

Source

fn build(vb: VarBuilder<'_>, config: Self::Config) -> Self

Source

fn forward(&self, input: &Self::Input) -> Self::Output

A generalized forward function.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§