SYMBOL INDEX (386 symbols across 46 files)

FILE: dataset.py
  class MyDataset (line 15) | class MyDataset(Dataset):
    method __init__ (line 16) | def __init__(self):
    method __len__ (line 20) | def __len__(self):
    method __getitem__ (line 23) | def __getitem__(self, idx):
  class MyCelebA (line 27) | class MyCelebA(CelebA):
    method _check_integrity (line 35) | def _check_integrity(self) -> bool:
  class OxfordPets (line 40) | class OxfordPets(Dataset):
    method __init__ (line 44) | def __init__(self,
    method __len__ (line 55) | def __len__(self):
    method __getitem__ (line 58) | def __getitem__(self, idx):
  class VAEDataset (line 66) | class VAEDataset(LightningDataModule):
    method __init__ (line 81) | def __init__(
    method setup (line 100) | def setup(self, stage: Optional[str] = None) -> None:
    method train_dataloader (line 155) | def train_dataloader(self) -> DataLoader:
    method val_dataloader (line 164) | def val_dataloader(self) -> Union[DataLoader, List[DataLoader]]:
    method test_dataloader (line 173) | def test_dataloader(self) -> Union[DataLoader, List[DataLoader]]:

FILE: experiment.py
  class VAEXperiment (line 15) | class VAEXperiment(pl.LightningModule):
    method __init__ (line 17) | def __init__(self,
    method forward (line 31) | def forward(self, input: Tensor, **kwargs) -> Tensor:
    method training_step (line 34) | def training_step(self, batch, batch_idx, optimizer_idx = 0):
    method validation_step (line 48) | def validation_step(self, batch, batch_idx, optimizer_idx = 0):
    method on_validation_end (line 61) | def on_validation_end(self) -> None:
    method sample_images (line 64) | def sample_images(self):
    method configure_optimizers (line 92) | def configure_optimizers(self):

FILE: models/base.py
  class BaseVAE (line 5) | class BaseVAE(nn.Module):
    method __init__ (line 7) | def __init__(self) -> None:
    method encode (line 10) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 13) | def decode(self, input: Tensor) -> Any:
    method sample (line 16) | def sample(self, batch_size:int, current_device: int, **kwargs) -> Ten...
    method generate (line 19) | def generate(self, x: Tensor, **kwargs) -> Tensor:
    method forward (line 23) | def forward(self, *inputs: Tensor) -> Tensor:
    method loss_function (line 27) | def loss_function(self, *inputs: Any, **kwargs) -> Tensor:

FILE: models/beta_vae.py
  class BetaVAE (line 8) | class BetaVAE(BaseVAE):
    method __init__ (line 12) | def __init__(self,
    method encode (line 88) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 105) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 112) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 124) | def forward(self, input: Tensor, **kwargs) -> Tensor:
    method loss_function (line 129) | def loss_function(self,
    method sample (line 154) | def sample(self,
    method generate (line 172) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/betatc_vae.py
  class BetaTCVAE (line 9) | class BetaTCVAE(BaseVAE):
    method __init__ (line 12) | def __init__(self,
    method encode (line 84) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 102) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 115) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 127) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method log_density_gaussian (line 132) | def log_density_gaussian(self, x: Tensor, mu: Tensor, logvar: Tensor):
    method loss_function (line 144) | def loss_function(self,
    method sample (line 213) | def sample(self,
    method generate (line 231) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/cat_vae.py
  class CategoricalVAE (line 9) | class CategoricalVAE(BaseVAE):
    method __init__ (line 11) | def __init__(self,
    method encode (line 89) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 105) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 118) | def reparameterize(self, z: Tensor, eps:float = 1e-7) -> Tensor:
    method forward (line 134) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 139) | def loss_function(self,
    method sample (line 179) | def sample(self,
    method generate (line 202) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/cvae.py
  class ConditionalVAE (line 8) | class ConditionalVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 83) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 100) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 107) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 119) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 133) | def loss_function(self,
    method sample (line 149) | def sample(self,
    method generate (line 170) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/dfcvae.py
  class DFCVAE (line 9) | class DFCVAE(BaseVAE):
    method __init__ (line 11) | def __init__(self,
    method encode (line 90) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 107) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 120) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 132) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method extract_features (line 142) | def extract_features(self,
    method loss_function (line 163) | def loss_function(self,
    method sample (line 192) | def sample(self,
    method generate (line 210) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/dip_vae.py
  class DIPVAE (line 8) | class DIPVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 78) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 95) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 108) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 120) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 125) | def loss_function(self,
    method sample (line 166) | def sample(self,
    method generate (line 184) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/fvae.py
  class FactorVAE (line 8) | class FactorVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 92) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 109) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 122) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 134) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method permute_latent (line 139) | def permute_latent(self, z: Tensor) -> Tensor:
    method loss_function (line 151) | def loss_function(self,
    method sample (line 203) | def sample(self,
    method generate (line 221) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/gamma_vae.py
  class GammaVAE (line 10) | class GammaVAE(BaseVAE):
    method __init__ (line 12) | def __init__(self,
    method weight_init (line 85) | def weight_init(self):
    method encode (line 92) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 109) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 116) | def reparameterize(self, alpha: Tensor, beta: Tensor) -> Tensor:
    method h_func (line 137) | def h_func(self, alpha: Tensor, eps: Tensor) -> Tensor:
    method inv_h_func (line 148) | def inv_h_func(self, alpha: Tensor, z: Tensor) -> Tensor:
    method forward (line 158) | def forward(self, input: Tensor, **kwargs) -> Tensor:
    method I_function (line 168) | def I_function(self, a, b, c, d):
    method vae_gamma_kl_loss (line 171) | def vae_gamma_kl_loss(self, a, b, c, d):
    method loss_function (line 184) | def loss_function(self,
    method sample (line 214) | def sample(self,
    method generate (line 230) | def generate(self, x: Tensor, **kwargs) -> Tensor:
  function init_ (line 239) | def init_(m):

FILE: models/hvae.py
  class HVAE (line 8) | class HVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode_z2 (line 115) | def encode_z2(self, input: Tensor) -> List[Tensor]:
    method encode_z1 (line 132) | def encode_z1(self, input: Tensor, z2: Tensor) -> List[Tensor]:
    method encode (line 145) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 153) | def decode(self, input: Tensor) -> Tensor:
    method reparameterize (line 158) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 170) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 192) | def loss_function(self,
    method sample (line 233) | def sample(self, batch_size:int, current_device: int, **kwargs) -> Ten...
    method generate (line 252) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/info_vae.py
  class InfoVAE (line 8) | class InfoVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 88) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 104) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 111) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 123) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 128) | def loss_function(self,
    method compute_kernel (line 150) | def compute_kernel(self,
    method compute_rbf (line 178) | def compute_rbf(self,
    method compute_inv_mult_quad (line 195) | def compute_inv_mult_quad(self,
    method compute_mmd (line 218) | def compute_mmd(self, z: Tensor) -> Tensor:
    method sample (line 231) | def sample(self,
    method generate (line 249) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/iwae.py
  class IWAE (line 8) | class IWAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 78) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 95) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 111) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 121) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 129) | def loss_function(self,
    method sample (line 162) | def sample(self,
    method generate (line 180) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/joint_vae.py
  class JointVAE (line 9) | class JointVAE(BaseVAE):
    method __init__ (line 12) | def __init__(self,
    method encode (line 111) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 129) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 142) | def reparameterize(self,
    method forward (line 170) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 175) | def loss_function(self,
    method sample (line 236) | def sample(self,
    method generate (line 261) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/logcosh_vae.py
  class LogCoshVAE (line 8) | class LogCoshVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 78) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 95) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 108) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 120) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 125) | def loss_function(self,
    method sample (line 157) | def sample(self,
    method generate (line 175) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/lvae.py
  function conv_out_shape (line 9) | def conv_out_shape(img_size):
  class EncoderBlock (line 12) | class EncoderBlock(nn.Module):
    method __init__ (line 13) | def __init__(self,
    method forward (line 32) | def forward(self, input: Tensor) -> Tensor:
  class LadderBlock (line 43) | class LadderBlock(nn.Module):
    method __init__ (line 44) | def __init__(self,
    method forward (line 55) | def forward(self, z: Tensor) -> Tensor:
  class LVAE (line 62) | class LVAE(BaseVAE):
    method __init__ (line 64) | def __init__(self,
    method encode (line 134) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 151) | def decode(self, z: Tensor, post_params: List) -> Tuple:
    method merge_gauss (line 173) | def merge_gauss(self,
    method compute_kl_divergence (line 186) | def compute_kl_divergence(self, z: Tensor, q_params: Tuple, p_params: ...
    method reparameterize (line 197) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 209) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 218) | def loss_function(self,
    method sample (line 239) | def sample(self,
    method generate (line 264) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/miwae.py
  class MIWAE (line 9) | class MIWAE(BaseVAE):
    method __init__ (line 11) | def __init__(self,
    method encode (line 81) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 98) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 114) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 124) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 132) | def loss_function(self,
    method sample (line 166) | def sample(self,
    method generate (line 184) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/mssim_vae.py
  class MSSIMVAE (line 9) | class MSSIMVAE(BaseVAE):
    method __init__ (line 11) | def __init__(self,
    method encode (line 84) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 101) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 114) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 126) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 131) | def loss_function(self,
    method sample (line 155) | def sample(self,
    method generate (line 173) | def generate(self, x: Tensor, **kwargs) -> Tensor:
  class MSSIM (line 182) | class MSSIM(nn.Module):
    method __init__ (line 184) | def __init__(self,
    method gaussian_window (line 203) | def gaussian_window(self, window_size:int, sigma: float) -> Tensor:
    method create_window (line 208) | def create_window(self, window_size, in_channels):
    method ssim (line 214) | def ssim(self,
    method forward (line 250) | def forward(self, img1: Tensor, img2: Tensor) -> Tensor:

FILE: models/swae.py
  class SWAE (line 9) | class SWAE(BaseVAE):
    method __init__ (line 11) | def __init__(self,
    method encode (line 84) | def encode(self, input: Tensor) -> Tensor:
    method decode (line 99) | def decode(self, z: Tensor) -> Tensor:
    method forward (line 106) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 110) | def loss_function(self,
    method get_random_projections (line 129) | def get_random_projections(self, latent_dim: int, num_samples: int) ->...
    method compute_swd (line 151) | def compute_swd(self,
    method sample (line 181) | def sample(self,
    method generate (line 199) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/twostage_vae.py
  class TwoStageVAE (line 8) | class TwoStageVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 100) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 117) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 130) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 142) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 148) | def loss_function(self,
    method sample (line 172) | def sample(self,
    method generate (line 190) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/vampvae.py
  class VampVAE (line 8) | class VampVAE(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 82) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 99) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 106) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 118) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 123) | def loss_function(self,
    method sample (line 170) | def sample(self,
    method generate (line 188) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/vanilla_vae.py
  class VanillaVAE (line 8) | class VanillaVAE(BaseVAE):
    method __init__ (line 11) | def __init__(self,
    method encode (line 77) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 94) | def decode(self, z: Tensor) -> Tensor:
    method reparameterize (line 107) | def reparameterize(self, mu: Tensor, logvar: Tensor) -> Tensor:
    method forward (line 119) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 124) | def loss_function(self,
    method sample (line 148) | def sample(self,
    method generate (line 166) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/vq_vae.py
  class VectorQuantizer (line 7) | class VectorQuantizer(nn.Module):
    method __init__ (line 12) | def __init__(self,
    method forward (line 24) | def forward(self, latents: Tensor) -> Tensor:
  class ResidualLayer (line 57) | class ResidualLayer(nn.Module):
    method __init__ (line 59) | def __init__(self,
    method forward (line 69) | def forward(self, input: Tensor) -> Tensor:
  class VQVAE (line 73) | class VQVAE(BaseVAE):
    method __init__ (line 75) | def __init__(self,
    method encode (line 168) | def encode(self, input: Tensor) -> List[Tensor]:
    method decode (line 178) | def decode(self, z: Tensor) -> Tensor:
    method forward (line 189) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 194) | def loss_function(self,
    method sample (line 213) | def sample(self,
    method generate (line 218) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: models/wae_mmd.py
  class WAE_MMD (line 8) | class WAE_MMD(BaseVAE):
    method __init__ (line 10) | def __init__(self,
    method encode (line 81) | def encode(self, input: Tensor) -> Tensor:
    method decode (line 96) | def decode(self, z: Tensor) -> Tensor:
    method forward (line 103) | def forward(self, input: Tensor, **kwargs) -> List[Tensor]:
    method loss_function (line 107) | def loss_function(self,
    method compute_kernel (line 125) | def compute_kernel(self,
    method compute_rbf (line 153) | def compute_rbf(self,
    method compute_inv_mult_quad (line 170) | def compute_inv_mult_quad(self,
    method compute_mmd (line 193) | def compute_mmd(self, z: Tensor, reg_weight: float) -> Tensor:
    method sample (line 206) | def sample(self,
    method generate (line 224) | def generate(self, x: Tensor, **kwargs) -> Tensor:

FILE: tests/bvae.py
  class TestVAE (line 7) | class TestVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):

FILE: tests/test_betatcvae.py
  class TestBetaTCVAE (line 7) | class TestBetaTCVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 24) | def test_loss(self):
    method test_sample (line 31) | def test_sample(self):
    method test_generate (line 36) | def test_generate(self):

FILE: tests/test_cat_vae.py
  class TestVAE (line 7) | class TestVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):
    method test_sample (line 31) | def test_sample(self):

FILE: tests/test_dfc.py
  class TestDFCVAE (line 7) | class TestDFCVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 18) | def test_forward(self):
    method test_loss (line 25) | def test_loss(self):
    method test_sample (line 32) | def test_sample(self):

FILE: tests/test_dipvae.py
  class TestDIPVAE (line 7) | class TestDIPVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 24) | def test_loss(self):
    method test_sample (line 31) | def test_sample(self):
    method test_generate (line 36) | def test_generate(self):

FILE: tests/test_fvae.py
  class TestFAE (line 7) | class TestFAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 20) | def test_forward(self):
    method test_loss (line 27) | def test_loss(self):
    method test_optim (line 36) | def test_optim(self):
    method test_sample (line 40) | def test_sample(self):

FILE: tests/test_gvae.py
  class TestGammaVAE (line 7) | class TestGammaVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 18) | def test_forward(self):
    method test_loss (line 25) | def test_loss(self):
    method test_sample (line 32) | def test_sample(self):

FILE: tests/test_hvae.py
  class TestHVAE (line 7) | class TestHVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):

FILE: tests/test_iwae.py
  class TestIWAE (line 7) | class TestIWAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):
    method test_sample (line 30) | def test_sample(self):

FILE: tests/test_joint_Vae.py
  class TestVAE (line 7) | class TestVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):
    method test_sample (line 31) | def test_sample(self):

FILE: tests/test_logcosh.py
  class TestVAE (line 7) | class TestVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):

FILE: tests/test_lvae.py
  class TestLVAE (line 7) | class TestLVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 24) | def test_loss(self):
    method test_sample (line 31) | def test_sample(self):

FILE: tests/test_miwae.py
  class TestMIWAE (line 7) | class TestMIWAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):
    method test_sample (line 30) | def test_sample(self):
    method test_generate (line 35) | def test_generate(self):

FILE: tests/test_mssimvae.py
  class TestMSSIMVAE (line 7) | class TestMSSIMVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 18) | def test_forward(self):
    method test_loss (line 25) | def test_loss(self):
    method test_sample (line 32) | def test_sample(self):

FILE: tests/test_swae.py
  class TestSWAE (line 7) | class TestSWAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 12) | def test_summary(self):
    method test_forward (line 16) | def test_forward(self):
    method test_loss (line 22) | def test_loss(self):

FILE: tests/test_vae.py
  class TestVAE (line 7) | class TestVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):

FILE: tests/test_vq_vae.py
  class TestVQVAE (line 7) | class TestVQVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 24) | def test_loss(self):
    method test_sample (line 31) | def test_sample(self):
    method test_generate (line 36) | def test_generate(self):

FILE: tests/test_wae.py
  class TestWAE (line 7) | class TestWAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 12) | def test_summary(self):
    method test_forward (line 16) | def test_forward(self):
    method test_loss (line 22) | def test_loss(self):

FILE: tests/text_cvae.py
  class TestCVAE (line 6) | class TestCVAE(unittest.TestCase):
    method setUp (line 8) | def setUp(self) -> None:
    method test_forward (line 12) | def test_forward(self):
    method test_loss (line 19) | def test_loss(self):

FILE: tests/text_vamp.py
  class TestVVAE (line 7) | class TestVVAE(unittest.TestCase):
    method setUp (line 9) | def setUp(self) -> None:
    method test_summary (line 13) | def test_summary(self):
    method test_forward (line 17) | def test_forward(self):
    method test_loss (line 23) | def test_loss(self):

FILE: utils.py
  function data_loader (line 8) | def data_loader(fn):