nessai.flows.fastslowflow
Fast-slow as described in Moss 2019 and based in the implementation in nnest. See: https://github.com/adammoss/nnest/blob/master/nnest/networks.py
Module Contents
Classes
- class nessai.flows.fastslowflow.FastSlowFlow(n_fast, n_slow, fast_class, slow_class, joint_class, fast_config=None, slow_config=None)
Bases:
nessai.flows.base.BaseFlow
- forward(self, x, context=None)
Apply the forward transformation and return samples in the latent space and the log-Jacobian determinant.
- Returns
torch.Tensor
Tensor of samples in the latent space
torch.Tensor
Tensor of log determinants of the Jacobian of the forward transformation