cascade.nn.Bilinear

class cascade.nn.Bilinear(n_vars, n_particles, eidx, tau=10.0, emb_dim=None)

Bases: Scaffold

Bilinearly parameterized edge logits

Parameters:

• n_vars (int) – Number of variables in the graph

• n_particles (int) – Number of SVGD particles

• eidx (LongTensor) – Scaffold edge indices of shape (2, n_edges)

• tau (float) – Gumbel-sigmoid temperature

• emb_dim (int) – Dimension of the bilinear parameterization

Methods

compute_logit

Attributes

INIT_STD