:py:mod:`nessai.flows.fastslowflow`
===================================

.. py:module:: nessai.flows.fastslowflow

.. autoapi-nested-parse::

   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

   ..
       !! processed by numpydoc !!


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   nessai.flows.fastslowflow.FastSlowFlow




.. py:class:: FastSlowFlow(n_fast, n_slow, fast_class, slow_class, joint_class, fast_config=None, slow_config=None)

   Bases: :py:obj:`nessai.flows.base.BaseFlow`

   
















   ..
       !! processed by numpydoc !!
   .. py:method:: forward(self, x, context=None)

      
      Apply the forward transformation and return samples in the
      latent space and the log-Jacobian determinant.



      :Returns:

          :obj:`torch.Tensor`
              Tensor of samples in the latent space

          :obj:`torch.Tensor`
              Tensor of log determinants of the Jacobian of the forward
              transformation













      ..
          !! processed by numpydoc !!