dataeval.utils.torch.models.Encoder

class dataeval.utils.torch.models.Encoder(channels=3)

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

• state (Initialize internal Module)

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

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