dataeval.utils.torch.models.Decoder

class dataeval.utils.torch.models.Decoder(channels)

A simple decoder to be used in an autoencoder model.

This is the decoder used by the Autoencoder model.

Parameters:

• channels (int) – Number of output channels

• state (Initialize internal Module)

• ScriptModule. (shared by both nn.Module and)

forward(x)

Perform a forward pass through the decoder.

Parameters:

x (torch.Tensor) – The encoded tensor.

Returns:

The reconstructed output tensor.

Return type:

torch.Tensor