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

FastSlowFlow

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