Struct std::Autoencoder

#include <std/ml/autoencoder.h> source

struct Autoencoder {
    struct Mlp enc;
    struct Mlp dec;
    struct AutoencoderConfig cfg;
    struct Rng rng;
    float lastRecon;
    float lastPenalty;
}

Fields

§struct Mlp encsource

§struct Mlp decsource

§struct AutoencoderConfig cfgsource

§struct Rng rngsource

§float lastReconsource

last loss call's reconstruction term

§float lastPenaltysource

... and its sparsity term

Methods

§&FloatTensor encode(const &FloatTensor x)source · definition

[m, latent]

§&FloatTensor decode(const &FloatTensor z)source · definition

[m, inDim]

§&FloatTensor reconstruct(const &FloatTensor x)source · definition

§void collect_params(&Vec<&FloatTensor> out) constsource · definition

Parameterized (module.h).

§void collect_state(&Vec<&FloatTensor> out) constsource · definition

§void drop()source · definition

§void free() consumesource · definition

= drop()

§&FloatTensor loss(const &FloatTensor x, const &FloatTensor y)source · definition

§static struct Autoencoder new(unsigned long inDim, const &AutoencoderConfig cfg)source · definition