dataeval.utils.models.Autoencoder¶

class dataeval.utils.models.Autoencoder(input_shape)¶

An autoencoder model with a separate encoder and decoder.

Parameters:¶

input_shape : tuple[int, ...]¶

Shape of the input data in CHW format.

forward(x)¶

Perform a forward pass through the encoder and decoder.

Parameters:¶

x : torch.Tensor¶

Input tensor

Returns:¶

The reconstructed output tensor.

Return type:¶

torch.Tensor