Struct AutoEncoder

pub struct AutoEncoder { /* private fields */ }

Implementations

impl AutoEncoder

pub fn new(cfg: &Config, vb: VarBuilder<'_>) -> Result<Self>

pub fn encode(&self, xs: &Tensor) -> Result<Tensor>

pub fn decode(&self, xs: &Tensor) -> Result<Tensor>

Trait Implementations

impl Clone for AutoEncoder

fn clone(&self) -> AutoEncoder

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for AutoEncoder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Module for AutoEncoder

fn forward(&self, xs: &Tensor) -> Result<Tensor>