Struct Mlp2 

pub struct Mlp2 { /* private fields */ }

Multilayer perceptron that outputs two tensors of the same size.

Trait Implementations

impl SubModel for Mlp2

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

TODO: support activation_out

type Config = MlpConfig

Configuration from which SubModel is constructed.

type Input = Tensor

Input of the SubModel.

type Output = (Tensor, Tensor)

Output of the SubModel.

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

A generalized forward function.

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

Clones SubModel with VarStore.