n2j.models.flow
#
Credit to Miles Cranmer: MilesCranmer/easy_normalizing_flow
Module Contents#
Classes#
same as Linear except has a configurable mask on the weights |
|
x0 only depends on x0, etc |
|
Attributes#
- class n2j.models.flow.MaskedLinear(in_features, out_features, bias=True)[source]#
Bases:
torch.nn.Linear
same as Linear except has a configurable mask on the weights
- class n2j.models.flow.MADE(nin, hidden_sizes, nout, num_masks=1, natural_ordering=False)[source]#
Bases:
torch.nn.Module
- class n2j.models.flow.MAF(features, context, hidden=100, nlayers=1)[source]#
Bases:
torch.nn.Module
x0 only depends on x0, etc