.. Custom template for autosummary functions. Can be removed when
.. https://github.com/sphinx-doc/sphinx/issues/7912 is fixed.

dgs.utils.torchtools.torch\_memory\_analysis
============================================

.. currentmodule:: dgs.utils.torchtools

.. autofunction:: torch_memory_analysis