Trait BuildModule

pub trait BuildModule {
    type Module: Module;

    // Required method
    fn build_module(
        &self,
        in_dim: usize,
        out_dim: usize,
        device: Device,
    ) -> Self::Module;
}

Build a Module

Required Associated Types

type Module: Module

Required Methods

fn build_module( &self, in_dim: usize, out_dim: usize, device: Device, ) -> Self::Module

Build a new module instance.

Args

• in_dim - Number of input feature dimensions.
• out_dim - Number of output feature dimensions.
• device - Device on which to create the model tensors.

Implementations on Foreign Types

impl<T: BuildModule + ?Sized> BuildModule for &T

type Module = <T as BuildModule>::Module

fn build_module( &self, in_dim: usize, out_dim: usize, device: Device, ) -> Self::Module

impl<T: BuildModule + ?Sized> BuildModule for Box<T>

type Module = <T as BuildModule>::Module

fn build_module( &self, in_dim: usize, out_dim: usize, device: Device, ) -> Self::Module

Implementors

impl BuildModule for LinearConfig

type Module = Linear

impl BuildModule for MlpConfig

type Module = Mlp

impl<A, B> BuildModule for ChainConfig<A, B>

where A: BuildModule, A::Module: for<'a> ModuleExtras<'a>, B: BuildModule, B::Module: for<'a> ModuleExtras<'a>,

type Module = Chain<<A as BuildModule>::Module, <B as BuildModule>::Module>