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
-
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