nichecompass.modules.VGAEModuleMixin

class nichecompass.modules.VGAEModuleMixin

VGAE module mix in class containing universal VGAE module functionalities.

Methods table

reparameterize(mu, logstd)

Use reparameterization trick for latent space normal distribution.

Methods

VGAEModuleMixin.reparameterize(mu, logstd)

Use reparameterization trick for latent space normal distribution.

Parameters:

• mu (Tensor) – Expected values of the latent space distribution (dim: n_obs, n_gps).

• logstd (Tensor) – Log standard deviations of the latent space distribution (dim: n_obs, n_gps).

Return type:

Tensor

Returns:

rep: Reparameterized latent features (dim: n_obs, n_gps).