Skip to content

nexuml_library.layers.model.conv_autoencoder

nexuml_library.layers.model.conv_autoencoder

Convolutional autoencoder building blocks.

ConvolutionalEncoder

Bases: LayerDefinition

Encode a 2D tensor into a latent vector.

VariationalLatent

Bases: LayerDefinition

Variational bottleneck layer emitting latent sample, moments, and KL loss.

ConvolutionalDecoder

Bases: LayerDefinition

Decode a latent vector back into a 2D tensor.