gfn.gym.helpers.diffusion_utils
===============================

.. py:module:: gfn.gym.helpers.diffusion_utils


Functions
---------

.. autoapisummary::

   gfn.gym.helpers.diffusion_utils.sliced_log_reward
   gfn.gym.helpers.diffusion_utils.viz_2d_slice


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

.. py:function:: sliced_log_reward(x, target, dims)

.. py:function:: viz_2d_slice(ax, target, dims, samples, plot_border, alpha=0.5, n_contour_levels=50, grid_width_n_points=200, log_reward_clamp_min=-10000.0, use_log_reward=False, max_n_samples = None)