nessai.utils.torchutils
=======================

.. py:module:: nessai.utils.torchutils

.. autoapi-nested-parse::

   Utilities for configuring torch.

   ..
       !! processed by numpydoc !!


Functions
---------

.. autoapisummary::

   nessai.utils.torchutils.set_torch_default_dtype


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

.. py:function:: set_torch_default_dtype(dtype) -> torch.dtype

   
   Set the default dtype for torch tenors.

   If dtype is None, returns the default dtype.

   :Parameters:

       **dtype** : {"float32", "float64", torch.dtype}
           The new default dtype for torch.



   :Returns:

       torch.dtype
           The torch dtype used to set the default.











   ..
       !! processed by numpydoc !!