A simple encoder to be used in an autoencoder model.
This is the encoder used by the Autoencoder model.
- Parameters:
- channels : int, default 3
Number of input channels
-
forward(x)
Perform a forward pass through the encoder.
- Parameters:
- x : torch.Tensor
Input tensor
- Returns:
The encoded representation of the input tensor.
- Return type:
torch.Tensor