cyto_dl.models.vae.priors.identity_prior module

class cyto_dl.models.vae.priors.identity_prior.IdentityPrior(dimensionality: int, **kwargs)

Bases: Prior

Prior class that doesn’t contribute to KL loss.

Effectively a Dirac delta distribution given z.

forward(z, mode='kl', **kwargs)

property param_size