timesead.models.common.ae

Classes

`AE`	Simple AE Implementation

Module Contents

class timesead.models.common.ae.AE(encoder: torch.nn.Module, decoder: torch.nn.Module, return_latent: bool = False)

Bases: torch.nn.Module

Simple AE Implementation

Initialize internal Module state, shared by both nn.Module and ScriptModule.

Parameters:

encoder (torch.nn.Module)
decoder (torch.nn.Module)
return_latent (bool)

encoder

decoder

return_latent = False

forward(x: torch.Tensor) → torch.Tensor | Tuple[torch.Tensor, torch.Tensor]

Parameters:: x (torch.Tensor)
Return type:: Union[torch.Tensor, Tuple[torch.Tensor, torch.Tensor]]