Decoder

class aixd.mlmodel.architecture.decoders.Decoder(*args: Any, **kwargs: Any)[source]

Bases: Module

Decoder module of the conditional (variational) autoencoder.

Parameters:
  • in_heads (Dict[str, nn.Module]) – Dictionary of input heads, where the keys are strings and the values are PyTorch modules.

  • out_heads (Dict[str, nn.Module]) – Dictionary of output heads, where the keys are strings and the values are PyTorch modules.

  • layer_widths (List[int]) – List of integer values representing the widths of the hidden layers in the decoder.

  • latent_dim (int) – Integer value representing the latent dimension of the model.

  • activation (Union[nn.Module, str]) – nn.Module or string representing the activation function to be used in the hidden layers of the decoder.

Methods

forward

Forward pass of the decoder.

summary

Prints a summary of a PyTorch model, including the number of parameters, the layers, their names, and the dimensionality.