Struct border_tch_agent::mlp::Mlp

pub struct Mlp { /* private fields */ }

Multilayer perceptron.

Trait Implementations

impl SubModel for Mlp

type Config = MlpConfig

Configuration from which SubModel is constructed.

type Input = Tensor

Input of the SubModel.

type Output = Tensor

Output of the SubModel.

fn forward(&self, x: &Self::Input) -> Tensor

A generalized forward function.

fn build(var_store: &VarStore, config: Self::Config) -> Self

Builds SubModel with VarStore and SubModel::Config.

fn clone_with_var_store(&self, var_store: &VarStore) -> Self

Clones SubModel with VarStore.

impl SubModel2 for Mlp

type Config = MlpConfig

Configuration from which SubModel2 is constructed.

type Input1 = Tensor

Input of the SubModel2.

type Input2 = Tensor

Input of the SubModel2.

type Output = Tensor

Output of the SubModel2.

fn forward(&self, input1: &Self::Input1, input2: &Self::Input2) -> Self::Output

A generalized forward function.

fn build(var_store: &VarStore, config: Self::Config) -> Self

Builds SubModel2 with VarStore and SubModel2::Config.

fn clone_with_var_store(&self, var_store: &VarStore) -> Self

Clones SubModel2 with VarStore.