Struct AtariCnn 

pub struct AtariCnn { /* private fields */ }

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

Trait Implementations

impl SubModel for AtariCnn

type Config = AtariCnnConfig

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.