cascade.nn.NilLatent

class cascade.nn.NilLatent(n_particles, latent_dim, vmap, **kwargs)[source]

Bases: Latent

Nil interventional latent module that always outputs the standard normal

Parameters:

  • n_particles (int) – Number of SVGD particles

  • latent_dim (int) – Dimensionality of the latent variable

  • vmap (LongTensor) – Variable index mapping with the parent module CausalNetwork

Methods

forward