AttentionUNet
=============

.. currentmodule:: deeplay.components.diffusion.attention_unet

.. autoclass:: AttentionUNet
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~AttentionUNet.configure
      ~AttentionUNet.forward

   .. rubric:: Methods Documentation

   .. automethod:: configure
   .. automethod:: forward