pyvene.models.layers

pyvene.models.layers#

Classes

AutoencoderLayer(input_dim, latent_dim, **kwargs)

An autoencoder with a single-layer encoder and single-layer decoder.

AutoencoderLayerBase(*args, **kwargs)

An abstract base class that defines an interface of an autoencoder.

InverseRotateLayer(lin_layer)

The inverse of a given LinearLayer module.

LowRankRotateLayer(n, m[, init_orth])

A linear transformation with orthogonal initialization.

RotateLayer(n[, init_orth])

A linear transformation with orthogonal initialization.

SubspaceLowRankRotateLayer(n, m[, init_orth])

A linear transformation with orthogonal initialization with subspace.