nlgm.autoencoder Module#

class nlgm.autoencoder.Decoder(hidden_dim=20, latent_dim=2)[source]#
forward(z)[source]#

Forward pass of the decoder.

Args:

z (torch.Tensor): Encoded input tensor.

Returns:

torch.Tensor: Decoded output tensor.

class nlgm.autoencoder.Encoder(hidden_dim=20, latent_dim=2)[source]#
forward(x)[source]#

Forward pass of the encoder.

Args:

x (torch.Tensor): Input tensor.

Returns:

torch.Tensor: Encoded output tensor.

class nlgm.autoencoder.GeometricAutoencoder(signature, hidden_dim=20, latent_dim=2)[source]#
forward(x)[source]#

Forward pass of the geometric autoencoder.

Args:

x (torch.Tensor): Input tensor.

Returns:

torch.Tensor: Decoded output tensor.