Struct border_tch_agent::cnn::Cnn

pub struct Cnn { /* private fields */ }

Convolutional neural network, which has the same architecture of the DQN paper.

Trait Implementations

impl SubModel for Cnn

type Config = CnnConfig

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.