torch\_openreml.covariance.DiagonalMatrix
=========================================

.. currentmodule:: torch_openreml.covariance

.. autoclass:: DiagonalMatrix

   
   
   
   
   
   .. rubric:: Methods

   .. autosummary::
   
      ~DiagonalMatrix.__call__
      ~DiagonalMatrix.auto_grad
      ~DiagonalMatrix.check_params
      ~DiagonalMatrix.from_param_dict
      ~DiagonalMatrix.get_intermediates
      ~DiagonalMatrix.grad
      ~DiagonalMatrix.manual_grad
      ~DiagonalMatrix.map_theta_to_dv
      ~DiagonalMatrix.map_theta_to_v
      ~DiagonalMatrix.reset_intermediates
      ~DiagonalMatrix.set_intermediates
      ~DiagonalMatrix.set_no_grad
      ~DiagonalMatrix.to_param_dict
      ~DiagonalMatrix.trans_grad
      ~DiagonalMatrix.trans_params
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~DiagonalMatrix.no_grad_index
      ~DiagonalMatrix.num_params
      ~DiagonalMatrix.param_names
      ~DiagonalMatrix.repr_dict
      ~DiagonalMatrix.shape
      ~DiagonalMatrix.trans