gflownet.utils.molecule.distributions
=====================================

.. py:module:: gflownet.utils.molecule.distributions


Functions
---------

.. autoapisummary::

   gflownet.utils.molecule.distributions.get_mixture_of_projected_normals


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

.. py:function:: get_mixture_of_projected_normals(weights, concentrations)

   :param weights: torch.tensor of shape [*batch_shape, n_components]
   :param concentrations: torch.tensor of shape [*batch_shape, n_components, n_dim]