cyto_dl.models.vae.priors.abstract_prior module#

class cyto_dl.models.vae.priors.abstract_prior.Prior(dimensionality: int, **kwargs)[source]#

Bases: Module

forward(z, mode='kl', inference=False, **kwargs)[source]#
kl_divergence(z)[source]#
property param_size#
sample(z)[source]#