Home Explore Help
Sign In
cnheider
/
draugr
mirror of https://github.com/cnheider/draugr
Watch 1
Star 1
Fork 0
Files Issues 0 Wiki
Branch: master
Branches Tags
draugr
/
docs
/
github
/
html
/
generated
/
draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.html

draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.html 179 KB
Permalink History Raw

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580 
  1. 

  2. <!DOCTYPE html>
    3. 

  3. 

  4. <html lang="en">
    5. 

  5.   <head>
    6. 

  6.     <meta charset="utf-8" />
    7. 

  7.     <meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.17.1: http://docutils.sourceforge.net/" />
    8. 

  8. 

  9.     <title>draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP &#8212; Draugr 1.0.1 documentation</title>
    10. 

  10.     <link rel="stylesheet" type="text/css" href="../_static/pygments.css" />
    11. 

  11.     <link rel="stylesheet" type="text/css" href="../_static/alabaster.css" />
    12. 

  12.     <link rel="stylesheet" type="text/css" href="../_static/graphviz.css" />
    13. 

  13.     <script data-url_root="../" id="documentation_options" src="../_static/documentation_options.js"></script>
    14. 

  14.     <script src="../_static/jquery.js"></script>
    15. 

  15.     <script src="../_static/underscore.js"></script>
    16. 

  16.     <script src="../_static/_sphinx_javascript_frameworks_compat.js"></script>
    17. 

  17.     <script src="../_static/doctools.js"></script>
    18. 

  18.     <link rel="canonical" href="pything.github.io/draugr/generated/draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.html" />
    19. 

  19.     <link rel="index" title="Index" href="../genindex.html" />
    20. 

  20.     <link rel="search" title="Search" href="../search.html" />
    21. 

  21.     <link rel="next" title="draugr.torch_utilities.architectures.distributional.normal.MultipleNormalMLP" href="draugr.torch_utilities.architectures.distributional.normal.MultipleNormalMLP.html" />
    22. 

  22.     <link rel="prev" title="draugr.torch_utilities.architectures.distributional.normal.MultiDimensionalNormalMLP" href="draugr.torch_utilities.architectures.distributional.normal.MultiDimensionalNormalMLP.html" />
    23. 

  23.    
    24. 

  24.   <link rel="stylesheet" href="../_static/custom.css" type="text/css" />
    25. 

  25.   
    26. 

  26.   
    27. 

  27.   <meta name="viewport" content="width=device-width, initial-scale=0.9, maximum-scale=0.9" />
    28. 

  28. 

  29.   </head><body>
    30. 

  30.   
    31. 

  31. 

  32.     <div class="document">
    33. 

  33.       <div class="documentwrapper">
    34. 

  34.         <div class="bodywrapper">
    35. 

  35.           
    36. 

  36. 

  37.           <div class="body" role="main">
    38. 

  38.             
    39. 

  39.   <section id="draugr-torch-utilities-architectures-distributional-normal-multivariatenormalmlp">
    40. 

  40. <h1>draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP<a class="headerlink" href="#draugr-torch-utilities-architectures-distributional-normal-multivariatenormalmlp" title="Permalink to this heading">¶</a></h1>
    41. 

  41. <dl class="py class">
    42. 

  42. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP">
    43. 

  43. <em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">draugr.torch_utilities.architectures.distributional.normal.</span></span><span class="sig-name descname"><span class="pre">MultiVariateNormalMLP</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">output_shape:</span> <span class="pre">~typing.Sequence</span> <span class="pre">=</span> <span class="pre">(2</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">)</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mean_head_activation:</span> <span class="pre">~typing.Optional[callable]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">*</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_shape:</span> <span class="pre">~typing.Sequence[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">hidden_layers:</span> <span class="pre">~typing.Sequence[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">hidden_layer_activation:</span> <span class="pre">~torch.nn.modules.module.Module</span> <span class="pre">=</span> <span class="pre">ReLU()</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">output_activation:</span> <span class="pre">~torch.nn.modules.module.Module</span> <span class="pre">=</span> <span class="pre">Identity()</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_bias:</span> <span class="pre">bool</span> <span class="pre">=</span> <span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_dropout:</span> <span class="pre">bool</span> <span class="pre">=</span> <span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dropout_prob:</span> <span class="pre">float</span> <span class="pre">=</span> <span class="pre">0.2</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">auto_build_hidden_layers_if_none=True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_multiplier:</span> <span class="pre">int</span> <span class="pre">=</span> <span class="pre">32</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_layer_width=1000</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">output_multiplier:</span> <span class="pre">int</span> <span class="pre">=</span> <span class="pre">16</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">default_init:</span> <span class="pre">callable</span> <span class="pre">=</span> <span class="pre">&lt;function</span> <span class="pre">fan_in_init&gt;</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/draugr/torch_utilities/architectures/distributional/normal.html#MultiVariateNormalMLP"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP" title="Permalink to this definition">¶</a></dt>
    44. 

  44. <dd><p>Bases: <a class="reference internal" href="draugr.torch_utilities.architectures.mlp.MLP.html#draugr.torch_utilities.architectures.mlp.MLP" title="draugr.torch_utilities.architectures.mlp.MLP"><code class="xref py py-class docutils literal notranslate"><span class="pre">MLP</span></code></a></p>
    45. 

  45. <p>description</p>
    46. 

  46. <dl class="py method">
    47. 

  47. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.__init__">
    48. 

  48. <span class="sig-name descname"><span class="pre">__init__</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">output_shape:</span> <span class="pre">~typing.Sequence</span> <span class="pre">=</span> <span class="pre">(2</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">)</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">mean_head_activation:</span> <span class="pre">~typing.Optional[callable]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">*</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_shape:</span> <span class="pre">~typing.Sequence[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">hidden_layers:</span> <span class="pre">~typing.Sequence[int]</span> <span class="pre">=</span> <span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">hidden_layer_activation:</span> <span class="pre">~torch.nn.modules.module.Module</span> <span class="pre">=</span> <span class="pre">ReLU()</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">output_activation:</span> <span class="pre">~torch.nn.modules.module.Module</span> <span class="pre">=</span> <span class="pre">Identity()</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_bias:</span> <span class="pre">bool</span> <span class="pre">=</span> <span class="pre">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_dropout:</span> <span class="pre">bool</span> <span class="pre">=</span> <span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dropout_prob:</span> <span class="pre">float</span> <span class="pre">=</span> <span class="pre">0.2</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">auto_build_hidden_layers_if_none=True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_multiplier:</span> <span class="pre">int</span> <span class="pre">=</span> <span class="pre">32</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_layer_width=1000</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">output_multiplier:</span> <span class="pre">int</span> <span class="pre">=</span> <span class="pre">16</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">default_init:</span> <span class="pre">callable</span> <span class="pre">=</span> <span class="pre">&lt;function</span> <span class="pre">fan_in_init&gt;</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/draugr/torch_utilities/architectures/distributional/normal.html#MultiVariateNormalMLP.__init__"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.__init__" title="Permalink to this definition">¶</a></dt>
    49. 

  49. <dd><p>Initializes internal Module state, shared by both nn.Module and ScriptModule.</p>
    50. 

  50. </dd></dl>
    51. 

  51. 

  52. <p class="rubric">Methods</p>
    53. 

  53. <table class="autosummary longtable docutils align-default">
    54. 

  54. <colgroup>
    55. 

  55. <col style="width: 10%" />
    56. 

  56. <col style="width: 90%" />
    57. 

  57. </colgroup>
    58. 

  58. <tbody>
    59. 

  59. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.__init__" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.__init__"><code class="xref py py-obj docutils literal notranslate"><span class="pre">__init__</span></code></a>([output_shape, ...])</p></td>
    60. 

  60. <td><p>Initializes internal Module state, shared by both nn.Module and ScriptModule.</p></td>
    61. 

  61. </tr>
    62. 

  62. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module"><code class="xref py py-obj docutils literal notranslate"><span class="pre">add_module</span></code></a>(name, module)</p></td>
    63. 

  63. <td><p>Adds a child module to the current module.</p></td>
    64. 

  64. </tr>
    65. 

  65. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.apply" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.apply"><code class="xref py py-obj docutils literal notranslate"><span class="pre">apply</span></code></a>(fn)</p></td>
    66. 

  66. <td><p>Applies <code class="docutils literal notranslate"><span class="pre">fn</span></code> recursively to every submodule (as returned by <code class="docutils literal notranslate"><span class="pre">.children()</span></code>) as well as self.</p></td>
    67. 

  67. </tr>
    68. 

  68. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.bfloat16" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.bfloat16"><code class="xref py py-obj docutils literal notranslate"><span class="pre">bfloat16</span></code></a>()</p></td>
    69. 

  69. <td><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">bfloat16</span></code> datatype.</p></td>
    70. 

  70. </tr>
    71. 

  71. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.buffers" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.buffers"><code class="xref py py-obj docutils literal notranslate"><span class="pre">buffers</span></code></a>([recurse])</p></td>
    72. 

  72. <td><p>Returns an iterator over module buffers.</p></td>
    73. 

  73. </tr>
    74. 

  74. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.children" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.children"><code class="xref py py-obj docutils literal notranslate"><span class="pre">children</span></code></a>()</p></td>
    75. 

  75. <td><p>Returns an iterator over immediate children modules.</p></td>
    76. 

  76. </tr>
    77. 

  77. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.construct_progressive_hidden_layers" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.construct_progressive_hidden_layers"><code class="xref py py-obj docutils literal notranslate"><span class="pre">construct_progressive_hidden_layers</span></code></a>(...[, ...])</p></td>
    78. 

  78. <td><p><dl class="field-list simple">
    79. 

  79. <dt class="field-odd">param _input_shape</dt>
    80. 

  80. <dd class="field-odd"><p></p></dd>
    81. 

  81. </dl>
    82. 

  82. </p></td>
    83. 

  83. </tr>
    84. 

  84. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cpu" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cpu"><code class="xref py py-obj docutils literal notranslate"><span class="pre">cpu</span></code></a>()</p></td>
    85. 

  85. <td><p>Moves all model parameters and buffers to the CPU.</p></td>
    86. 

  86. </tr>
    87. 

  87. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda"><code class="xref py py-obj docutils literal notranslate"><span class="pre">cuda</span></code></a>([device])</p></td>
    88. 

  88. <td><p>Moves all model parameters and buffers to the GPU.</p></td>
    89. 

  89. </tr>
    90. 

  90. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.double" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.double"><code class="xref py py-obj docutils literal notranslate"><span class="pre">double</span></code></a>()</p></td>
    91. 

  91. <td><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">double</span></code> datatype.</p></td>
    92. 

  92. </tr>
    93. 

  93. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.eval" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.eval"><code class="xref py py-obj docutils literal notranslate"><span class="pre">eval</span></code></a>()</p></td>
    94. 

  94. <td><p>Sets the module in evaluation mode.</p></td>
    95. 

  95. </tr>
    96. 

  96. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.extra_repr" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.extra_repr"><code class="xref py py-obj docutils literal notranslate"><span class="pre">extra_repr</span></code></a>()</p></td>
    97. 

  97. <td><p>Set the extra representation of the module</p></td>
    98. 

  98. </tr>
    99. 

  99. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.float" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.float"><code class="xref py py-obj docutils literal notranslate"><span class="pre">float</span></code></a>()</p></td>
    100. 

  100. <td><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">float</span></code> datatype.</p></td>
    101. 

  101. </tr>
    102. 

  102. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward"><code class="xref py py-obj docutils literal notranslate"><span class="pre">forward</span></code></a>(*x[, min_std, max_std])</p></td>
    103. 

  103. <td><p><dl class="field-list simple">
    104. 

  104. <dt class="field-odd">param x</dt>
    105. 

  105. <dd class="field-odd"><p></p></dd>
    106. 

  106. </dl>
    107. 

  107. </p></td>
    108. 

  108. </tr>
    109. 

  109. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_buffer" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_buffer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_buffer</span></code></a>(target)</p></td>
    110. 

  110. <td><p>Returns the buffer given by <code class="docutils literal notranslate"><span class="pre">target</span></code> if it exists, otherwise throws an error.</p></td>
    111. 

  111. </tr>
    112. 

  112. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_extra_state" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_extra_state"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_extra_state</span></code></a>()</p></td>
    113. 

  113. <td><p>Returns any extra state to include in the module's state_dict.</p></td>
    114. 

  114. </tr>
    115. 

  115. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_parameter" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_parameter"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_parameter</span></code></a>(target)</p></td>
    116. 

  116. <td><p>Returns the parameter given by <code class="docutils literal notranslate"><span class="pre">target</span></code> if it exists, otherwise throws an error.</p></td>
    117. 

  117. </tr>
    118. 

  118. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_submodule" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_submodule"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_submodule</span></code></a>(target)</p></td>
    119. 

  119. <td><p>Returns the submodule given by <code class="docutils literal notranslate"><span class="pre">target</span></code> if it exists, otherwise throws an error.</p></td>
    120. 

  120. </tr>
    121. 

  121. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.half" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.half"><code class="xref py py-obj docutils literal notranslate"><span class="pre">half</span></code></a>()</p></td>
    122. 

  122. <td><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">half</span></code> datatype.</p></td>
    123. 

  123. </tr>
    124. 

  124. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_input_shape" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_input_shape"><code class="xref py py-obj docutils literal notranslate"><span class="pre">infer_input_shape</span></code></a>(input_shape)</p></td>
    125. 

  125. <td><p><dl class="field-list simple">
    126. 

  126. <dt class="field-odd">return</dt>
    127. 

  127. <dd class="field-odd"><p></p></dd>
    128. 

  128. </dl>
    129. 

  129. </p></td>
    130. 

  130. </tr>
    131. 

  131. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_output_shape" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_output_shape"><code class="xref py py-obj docutils literal notranslate"><span class="pre">infer_output_shape</span></code></a>(output_shape)</p></td>
    132. 

  132. <td><p><dl class="field-list simple">
    133. 

  133. <dt class="field-odd">return</dt>
    134. 

  134. <dd class="field-odd"><p></p></dd>
    135. 

  135. </dl>
    136. 

  136. </p></td>
    137. 

  137. </tr>
    138. 

  138. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.ipu" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.ipu"><code class="xref py py-obj docutils literal notranslate"><span class="pre">ipu</span></code></a>([device])</p></td>
    139. 

  139. <td><p>Moves all model parameters and buffers to the IPU.</p></td>
    140. 

  140. </tr>
    141. 

  141. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict"><code class="xref py py-obj docutils literal notranslate"><span class="pre">load_state_dict</span></code></a>(state_dict[, strict])</p></td>
    142. 

  142. <td><p>Copies parameters and buffers from <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a> into this module and its descendants.</p></td>
    143. 

  143. </tr>
    144. 

  144. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.modules" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.modules"><code class="xref py py-obj docutils literal notranslate"><span class="pre">modules</span></code></a>()</p></td>
    145. 

  145. <td><p>Returns an iterator over all modules in the network.</p></td>
    146. 

  146. </tr>
    147. 

  147. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_buffers" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_buffers"><code class="xref py py-obj docutils literal notranslate"><span class="pre">named_buffers</span></code></a>([prefix, recurse])</p></td>
    148. 

  148. <td><p>Returns an iterator over module buffers, yielding both the name of the buffer as well as the buffer itself.</p></td>
    149. 

  149. </tr>
    150. 

  150. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_children" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_children"><code class="xref py py-obj docutils literal notranslate"><span class="pre">named_children</span></code></a>()</p></td>
    151. 

  151. <td><p>Returns an iterator over immediate children modules, yielding both the name of the module as well as the module itself.</p></td>
    152. 

  152. </tr>
    153. 

  153. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_modules" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_modules"><code class="xref py py-obj docutils literal notranslate"><span class="pre">named_modules</span></code></a>([memo, prefix, remove_duplicate])</p></td>
    154. 

  154. <td><p>Returns an iterator over all modules in the network, yielding both the name of the module as well as the module itself.</p></td>
    155. 

  155. </tr>
    156. 

  156. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_parameters" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_parameters"><code class="xref py py-obj docutils literal notranslate"><span class="pre">named_parameters</span></code></a>([prefix, recurse])</p></td>
    157. 

  157. <td><p>Returns an iterator over module parameters, yielding both the name of the parameter as well as the parameter itself.</p></td>
    158. 

  158. </tr>
    159. 

  159. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.parameters" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.parameters"><code class="xref py py-obj docutils literal notranslate"><span class="pre">parameters</span></code></a>([recurse])</p></td>
    160. 

  160. <td><p>Returns an iterator over module parameters.</p></td>
    161. 

  161. </tr>
    162. 

  162. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_backward_hook" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_backward_hook"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_backward_hook</span></code></a>(hook)</p></td>
    163. 

  163. <td><p>Registers a backward hook on the module.</p></td>
    164. 

  164. </tr>
    165. 

  165. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_buffer" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_buffer"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_buffer</span></code></a>(name, tensor[, persistent])</p></td>
    166. 

  166. <td><p>Adds a buffer to the module.</p></td>
    167. 

  167. </tr>
    168. 

  168. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_hook" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_hook"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_forward_hook</span></code></a>(hook)</p></td>
    169. 

  169. <td><p>Registers a forward hook on the module.</p></td>
    170. 

  170. </tr>
    171. 

  171. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_pre_hook" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_pre_hook"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_forward_pre_hook</span></code></a>(hook)</p></td>
    172. 

  172. <td><p>Registers a forward pre-hook on the module.</p></td>
    173. 

  173. </tr>
    174. 

  174. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_full_backward_hook" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_full_backward_hook"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_full_backward_hook</span></code></a>(hook)</p></td>
    175. 

  175. <td><p>Registers a backward hook on the module.</p></td>
    176. 

  176. </tr>
    177. 

  177. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_load_state_dict_post_hook" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_load_state_dict_post_hook"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_load_state_dict_post_hook</span></code></a>(hook)</p></td>
    178. 

  178. <td><p>Registers a post hook to be run after module's <code class="docutils literal notranslate"><span class="pre">load_state_dict</span></code> is called.</p></td>
    179. 

  179. </tr>
    180. 

  180. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_module" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_module"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_module</span></code></a>(name, module)</p></td>
    181. 

  181. <td><p>Alias for <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module"><code class="xref py py-func docutils literal notranslate"><span class="pre">add_module()</span></code></a>.</p></td>
    182. 

  182. </tr>
    183. 

  183. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_parameter" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_parameter"><code class="xref py py-obj docutils literal notranslate"><span class="pre">register_parameter</span></code></a>(name, param)</p></td>
    184. 

  184. <td><p>Adds a parameter to the module.</p></td>
    185. 

  185. </tr>
    186. 

  186. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.requires_grad_" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.requires_grad_"><code class="xref py py-obj docutils literal notranslate"><span class="pre">requires_grad_</span></code></a>([requires_grad])</p></td>
    187. 

  187. <td><p>Change if autograd should record operations on parameters in this module.</p></td>
    188. 

  188. </tr>
    189. 

  189. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.sample_input" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.sample_input"><code class="xref py py-obj docutils literal notranslate"><span class="pre">sample_input</span></code></a>()</p></td>
    190. 

  190. <td><p><dl class="field-list simple">
    191. 

  191. <dt class="field-odd">return</dt>
    192. 

  192. <dd class="field-odd"><p></p></dd>
    193. 

  193. </dl>
    194. 

  194. </p></td>
    195. 

  195. </tr>
    196. 

  196. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.set_extra_state" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.set_extra_state"><code class="xref py py-obj docutils literal notranslate"><span class="pre">set_extra_state</span></code></a>(state)</p></td>
    197. 

  197. <td><p>This function is called from <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict"><code class="xref py py-func docutils literal notranslate"><span class="pre">load_state_dict()</span></code></a> to handle any extra state found within the <cite>state_dict</cite>.</p></td>
    198. 

  198. </tr>
    199. 

  199. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.share_memory" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.share_memory"><code class="xref py py-obj docutils literal notranslate"><span class="pre">share_memory</span></code></a>()</p></td>
    200. 

  200. <td><p>See <code class="xref py py-meth docutils literal notranslate"><span class="pre">torch.Tensor.share_memory_()</span></code></p></td>
    201. 

  201. </tr>
    202. 

  202. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-obj docutils literal notranslate"><span class="pre">state_dict</span></code></a>(*args[, destination, prefix, ...])</p></td>
    203. 

  203. <td><p>Returns a dictionary containing a whole state of the module.</p></td>
    204. 

  204. </tr>
    205. 

  205. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to"><code class="xref py py-obj docutils literal notranslate"><span class="pre">to</span></code></a>(*args, **kwargs)</p></td>
    206. 

  206. <td><p>Moves and/or casts the parameters and buffers.</p></td>
    207. 

  207. </tr>
    208. 

  208. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to_empty" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to_empty"><code class="xref py py-obj docutils literal notranslate"><span class="pre">to_empty</span></code></a>(*, device)</p></td>
    209. 

  209. <td><p>Moves the parameters and buffers to the specified device without copying storage.</p></td>
    210. 

  210. </tr>
    211. 

  211. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.train" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.train"><code class="xref py py-obj docutils literal notranslate"><span class="pre">train</span></code></a>([mode])</p></td>
    212. 

  212. <td><p>Sets the module in training mode.</p></td>
    213. 

  213. </tr>
    214. 

  214. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.type" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.type"><code class="xref py py-obj docutils literal notranslate"><span class="pre">type</span></code></a>(dst_type)</p></td>
    215. 

  215. <td><p>Casts all parameters and buffers to <code class="xref py py-attr docutils literal notranslate"><span class="pre">dst_type</span></code>.</p></td>
    216. 

  216. </tr>
    217. 

  217. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.xpu" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.xpu"><code class="xref py py-obj docutils literal notranslate"><span class="pre">xpu</span></code></a>([device])</p></td>
    218. 

  218. <td><p>Moves all model parameters and buffers to the XPU.</p></td>
    219. 

  219. </tr>
    220. 

  220. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.zero_grad" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.zero_grad"><code class="xref py py-obj docutils literal notranslate"><span class="pre">zero_grad</span></code></a>([set_to_none])</p></td>
    221. 

  221. <td><p>Sets gradients of all model parameters to zero.</p></td>
    222. 

  222. </tr>
    223. 

  223. </tbody>
    224. 

  224. </table>
    225. 

  225. <p class="rubric">Attributes</p>
    226. 

  226. <table class="autosummary longtable docutils align-default">
    227. 

  227. <colgroup>
    228. 

  228. <col style="width: 10%" />
    229. 

  229. <col style="width: 90%" />
    230. 

  230. </colgroup>
    231. 

  231. <tbody>
    232. 

  232. <tr class="row-odd"><td><p><code class="xref py py-obj docutils literal notranslate"><span class="pre">T_destination</span></code></p></td>
    233. 

  233. <td><p>alias of TypeVar('T_destination', bound=<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Dict" title="(in Python v3.10)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Dict</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Any" title="(in Python v3.10)"><code class="xref py py-obj docutils literal notranslate"><span class="pre">Any</span></code></a>])</p></td>
    234. 

  234. </tr>
    235. 

  235. <tr class="row-even"><td><p><code class="xref py py-obj docutils literal notranslate"><span class="pre">dump_patches</span></code></p></td>
    236. 

  236. <td><p></p></td>
    237. 

  237. </tr>
    238. 

  238. <tr class="row-odd"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.input_shape" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.input_shape"><code class="xref py py-obj docutils literal notranslate"><span class="pre">input_shape</span></code></a></p></td>
    239. 

  239. <td><p>return: :rtype:</p></td>
    240. 

  240. </tr>
    241. 

  241. <tr class="row-even"><td><p><a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.output_shape" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.output_shape"><code class="xref py py-obj docutils literal notranslate"><span class="pre">output_shape</span></code></a></p></td>
    242. 

  242. <td><p>return: :rtype:</p></td>
    243. 

  243. </tr>
    244. 

  244. </tbody>
    245. 

  245. </table>
    246. 

  246. <dl class="py method">
    247. 

  247. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module">
    248. 

  248. <span class="sig-name descname"><span class="pre">add_module</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">module</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module" title="Permalink to this definition">¶</a></dt>
    249. 

  249. <dd><p>Adds a child module to the current module.</p>
    250. 

  250. <p>The module can be accessed as an attribute using the given name.</p>
    251. 

  251. <dl class="field-list simple">
    252. 

  252. <dt class="field-odd">Parameters</dt>
    253. 

  253. <dd class="field-odd"><ul class="simple">
    254. 

  254. <li><p><strong>name</strong> (<em>string</em>) – name of the child module. The child module can be
    255. 

  255. accessed from this module using the given name</p></li>
    256. 

  256. <li><p><strong>module</strong> (<em>Module</em>) – child module to be added to the module.</p></li>
    257. 

  257. </ul>
    258. 

  258. </dd>
    259. 

  259. </dl>
    260. 

  260. </dd></dl>
    261. 

  261. 

  262. <dl class="py method">
    263. 

  263. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.apply">
    264. 

  264. <span class="sig-name descname"><span class="pre">apply</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">fn</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Callable" title="(in Python v3.10)"><span class="pre">Callable</span></a><span class="p"><span class="pre">[</span></span><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.apply" title="Permalink to this definition">¶</a></dt>
    265. 

  265. <dd><p>Applies <code class="docutils literal notranslate"><span class="pre">fn</span></code> recursively to every submodule (as returned by <code class="docutils literal notranslate"><span class="pre">.children()</span></code>)
    266. 

  266. as well as self. Typical use includes initializing the parameters of a model
    267. 

  267. (see also <span class="xref std std-ref">nn-init-doc</span>).</p>
    268. 

  268. <dl class="field-list simple">
    269. 

  269. <dt class="field-odd">Parameters</dt>
    270. 

  270. <dd class="field-odd"><p><strong>fn</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">Module</span></code> -&gt; None) – function to be applied to each submodule</p>
    271. 

  271. </dd>
    272. 

  272. <dt class="field-even">Returns</dt>
    273. 

  273. <dd class="field-even"><p>self</p>
    274. 

  274. </dd>
    275. 

  275. <dt class="field-odd">Return type</dt>
    276. 

  276. <dd class="field-odd"><p>Module</p>
    277. 

  277. </dd>
    278. 

  278. </dl>
    279. 

  279. <p>Example:</p>
    280. 

  280. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="nd">@torch</span><span class="o">.</span><span class="n">no_grad</span><span class="p">()</span>
    281. 

  281. <span class="gp">&gt;&gt;&gt; </span><span class="k">def</span> <span class="nf">init_weights</span><span class="p">(</span><span class="n">m</span><span class="p">):</span>
    282. 

  282. <span class="gp">&gt;&gt;&gt; </span>    <span class="nb">print</span><span class="p">(</span><span class="n">m</span><span class="p">)</span>
    283. 

  283. <span class="gp">&gt;&gt;&gt; </span>    <span class="k">if</span> <span class="nb">type</span><span class="p">(</span><span class="n">m</span><span class="p">)</span> <span class="o">==</span> <span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">:</span>
    284. 

  284. <span class="gp">&gt;&gt;&gt; </span>        <span class="n">m</span><span class="o">.</span><span class="n">weight</span><span class="o">.</span><span class="n">fill_</span><span class="p">(</span><span class="mf">1.0</span><span class="p">)</span>
    285. 

  285. <span class="gp">&gt;&gt;&gt; </span>        <span class="nb">print</span><span class="p">(</span><span class="n">m</span><span class="o">.</span><span class="n">weight</span><span class="p">)</span>
    286. 

  286. <span class="gp">&gt;&gt;&gt; </span><span class="n">net</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Sequential</span><span class="p">(</span><span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span><span class="p">),</span> <span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span><span class="p">))</span>
    287. 

  287. <span class="gp">&gt;&gt;&gt; </span><span class="n">net</span><span class="o">.</span><span class="n">apply</span><span class="p">(</span><span class="n">init_weights</span><span class="p">)</span>
    288. 

  288. <span class="go">Linear(in_features=2, out_features=2, bias=True)</span>
    289. 

  289. <span class="go">Parameter containing:</span>
    290. 

  290. <span class="go">tensor([[ 1.,  1.],</span>
    291. 

  291. <span class="go">        [ 1.,  1.]])</span>
    292. 

  292. <span class="go">Linear(in_features=2, out_features=2, bias=True)</span>
    293. 

  293. <span class="go">Parameter containing:</span>
    294. 

  294. <span class="go">tensor([[ 1.,  1.],</span>
    295. 

  295. <span class="go">        [ 1.,  1.]])</span>
    296. 

  296. <span class="go">Sequential(</span>
    297. 

  297. <span class="go">  (0): Linear(in_features=2, out_features=2, bias=True)</span>
    298. 

  298. <span class="go">  (1): Linear(in_features=2, out_features=2, bias=True)</span>
    299. 

  299. <span class="go">)</span>
    300. 

  300. <span class="go">Sequential(</span>
    301. 

  301. <span class="go">  (0): Linear(in_features=2, out_features=2, bias=True)</span>
    302. 

  302. <span class="go">  (1): Linear(in_features=2, out_features=2, bias=True)</span>
    303. 

  303. <span class="go">)</span>
    304. 

  304. </pre></div>
    305. 

  305. </div>
    306. 

  306. </dd></dl>
    307. 

  307. 

  308. <dl class="py method">
    309. 

  309. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.bfloat16">
    310. 

  310. <span class="sig-name descname"><span class="pre">bfloat16</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.bfloat16" title="Permalink to this definition">¶</a></dt>
    311. 

  311. <dd><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">bfloat16</span></code> datatype.</p>
    312. 

  312. <div class="admonition note">
    313. 

  313. <p class="admonition-title">Note</p>
    314. 

  314. <p>This method modifies the module in-place.</p>
    315. 

  315. </div>
    316. 

  316. <dl class="field-list simple">
    317. 

  317. <dt class="field-odd">Returns</dt>
    318. 

  318. <dd class="field-odd"><p>self</p>
    319. 

  319. </dd>
    320. 

  320. <dt class="field-even">Return type</dt>
    321. 

  321. <dd class="field-even"><p>Module</p>
    322. 

  322. </dd>
    323. 

  323. </dl>
    324. 

  324. </dd></dl>
    325. 

  325. 

  326. <dl class="py method">
    327. 

  327. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.buffers">
    328. 

  328. <span class="sig-name descname"><span class="pre">buffers</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">recurse</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Iterator" title="(in Python v3.10)"><span class="pre">Iterator</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.buffers" title="Permalink to this definition">¶</a></dt>
    329. 

  329. <dd><p>Returns an iterator over module buffers.</p>
    330. 

  330. <dl class="field-list simple">
    331. 

  331. <dt class="field-odd">Parameters</dt>
    332. 

  332. <dd class="field-odd"><p><strong>recurse</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – if True, then yields buffers of this module
    333. 

  333. and all submodules. Otherwise, yields only buffers that
    334. 

  334. are direct members of this module.</p>
    335. 

  335. </dd>
    336. 

  336. <dt class="field-even">Yields</dt>
    337. 

  337. <dd class="field-even"><p><em>torch.Tensor</em> – module buffer</p>
    338. 

  338. </dd>
    339. 

  339. </dl>
    340. 

  340. <p>Example:</p>
    341. 

  341. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="k">for</span> <span class="n">buf</span> <span class="ow">in</span> <span class="n">model</span><span class="o">.</span><span class="n">buffers</span><span class="p">():</span>
    342. 

  342. <span class="gp">&gt;&gt;&gt; </span>    <span class="nb">print</span><span class="p">(</span><span class="nb">type</span><span class="p">(</span><span class="n">buf</span><span class="p">),</span> <span class="n">buf</span><span class="o">.</span><span class="n">size</span><span class="p">())</span>
    343. 

  343. <span class="go">&lt;class &#39;torch.Tensor&#39;&gt; (20L,)</span>
    344. 

  344. <span class="go">&lt;class &#39;torch.Tensor&#39;&gt; (20L, 1L, 5L, 5L)</span>
    345. 

  345. </pre></div>
    346. 

  346. </div>
    347. 

  347. </dd></dl>
    348. 

  348. 

  349. <dl class="py method">
    350. 

  350. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.children">
    351. 

  351. <span class="sig-name descname"><span class="pre">children</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Iterator" title="(in Python v3.10)"><span class="pre">Iterator</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.children" title="Permalink to this definition">¶</a></dt>
    352. 

  352. <dd><p>Returns an iterator over immediate children modules.</p>
    353. 

  353. <dl class="field-list simple">
    354. 

  354. <dt class="field-odd">Yields</dt>
    355. 

  355. <dd class="field-odd"><p><em>Module</em> – a child module</p>
    356. 

  356. </dd>
    357. 

  357. </dl>
    358. 

  358. </dd></dl>
    359. 

  359. 

  360. <dl class="py method">
    361. 

  361. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.construct_progressive_hidden_layers">
    362. 

  362. <em class="property"><span class="pre">static</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">construct_progressive_hidden_layers</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">_input_shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">_output_shape</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">input_multiplier</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#float" title="(in Python v3.10)"><span class="pre">float</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">32</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">output_multiplier</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#float" title="(in Python v3.10)"><span class="pre">float</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">16</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_layer_width</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">1000</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Sequence" title="(in Python v3.10)"><span class="pre">Sequence</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.construct_progressive_hidden_layers" title="Permalink to this definition">¶</a></dt>
    363. 

  363. <dd><dl class="field-list simple">
    364. 

  364. <dt class="field-odd">Parameters</dt>
    365. 

  365. <dd class="field-odd"><ul class="simple">
    366. 

  366. <li><p><strong>_input_shape</strong> – </p></li>
    367. 

  367. <li><p><strong>_output_shape</strong> – </p></li>
    368. 

  368. <li><p><strong>input_multiplier</strong> – </p></li>
    369. 

  369. <li><p><strong>output_multiplier</strong> – </p></li>
    370. 

  370. <li><p><strong>max_layer_width</strong> – </p></li>
    371. 

  371. </ul>
    372. 

  372. </dd>
    373. 

  373. <dt class="field-even">Returns</dt>
    374. 

  374. <dd class="field-even"><p></p>
    375. 

  375. </dd>
    376. 

  376. <dt class="field-odd">Return type</dt>
    377. 

  377. <dd class="field-odd"><p></p>
    378. 

  378. </dd>
    379. 

  379. </dl>
    380. 

  380. </dd></dl>
    381. 

  381. 

  382. <dl class="py method">
    383. 

  383. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cpu">
    384. 

  384. <span class="sig-name descname"><span class="pre">cpu</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cpu" title="Permalink to this definition">¶</a></dt>
    385. 

  385. <dd><p>Moves all model parameters and buffers to the CPU.</p>
    386. 

  386. <div class="admonition note">
    387. 

  387. <p class="admonition-title">Note</p>
    388. 

  388. <p>This method modifies the module in-place.</p>
    389. 

  389. </div>
    390. 

  390. <dl class="field-list simple">
    391. 

  391. <dt class="field-odd">Returns</dt>
    392. 

  392. <dd class="field-odd"><p>self</p>
    393. 

  393. </dd>
    394. 

  394. <dt class="field-even">Return type</dt>
    395. 

  395. <dd class="field-even"><p>Module</p>
    396. 

  396. </dd>
    397. 

  397. </dl>
    398. 

  398. </dd></dl>
    399. 

  399. 

  400. <dl class="py method">
    401. 

  401. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda">
    402. 

  402. <span class="sig-name descname"><span class="pre">cuda</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">device</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda" title="Permalink to this definition">¶</a></dt>
    403. 

  403. <dd><p>Moves all model parameters and buffers to the GPU.</p>
    404. 

  404. <p>This also makes associated parameters and buffers different objects. So
    405. 

  405. it should be called before constructing optimizer if the module will
    406. 

  406. live on GPU while being optimized.</p>
    407. 

  407. <div class="admonition note">
    408. 

  408. <p class="admonition-title">Note</p>
    409. 

  409. <p>This method modifies the module in-place.</p>
    410. 

  410. </div>
    411. 

  411. <dl class="field-list simple">
    412. 

  412. <dt class="field-odd">Parameters</dt>
    413. 

  413. <dd class="field-odd"><p><strong>device</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><em>int</em></a><em>, </em><em>optional</em>) – if specified, all parameters will be
    414. 

  414. copied to that device</p>
    415. 

  415. </dd>
    416. 

  416. <dt class="field-even">Returns</dt>
    417. 

  417. <dd class="field-even"><p>self</p>
    418. 

  418. </dd>
    419. 

  419. <dt class="field-odd">Return type</dt>
    420. 

  420. <dd class="field-odd"><p>Module</p>
    421. 

  421. </dd>
    422. 

  422. </dl>
    423. 

  423. </dd></dl>
    424. 

  424. 

  425. <dl class="py method">
    426. 

  426. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.double">
    427. 

  427. <span class="sig-name descname"><span class="pre">double</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.double" title="Permalink to this definition">¶</a></dt>
    428. 

  428. <dd><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">double</span></code> datatype.</p>
    429. 

  429. <div class="admonition note">
    430. 

  430. <p class="admonition-title">Note</p>
    431. 

  431. <p>This method modifies the module in-place.</p>
    432. 

  432. </div>
    433. 

  433. <dl class="field-list simple">
    434. 

  434. <dt class="field-odd">Returns</dt>
    435. 

  435. <dd class="field-odd"><p>self</p>
    436. 

  436. </dd>
    437. 

  437. <dt class="field-even">Return type</dt>
    438. 

  438. <dd class="field-even"><p>Module</p>
    439. 

  439. </dd>
    440. 

  440. </dl>
    441. 

  441. </dd></dl>
    442. 

  442. 

  443. <dl class="py method">
    444. 

  444. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.eval">
    445. 

  445. <span class="sig-name descname"><span class="pre">eval</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.eval" title="Permalink to this definition">¶</a></dt>
    446. 

  446. <dd><p>Sets the module in evaluation mode.</p>
    447. 

  447. <p>This has any effect only on certain modules. See documentations of
    448. 

  448. particular modules for details of their behaviors in training/evaluation
    449. 

  449. mode, if they are affected, e.g. <code class="xref py py-class docutils literal notranslate"><span class="pre">Dropout</span></code>, <code class="xref py py-class docutils literal notranslate"><span class="pre">BatchNorm</span></code>,
    450. 

  450. etc.</p>
    451. 

  451. <p>This is equivalent with <code class="xref py py-meth docutils literal notranslate"><span class="pre">self.train(False)</span></code>.</p>
    452. 

  452. <p>See <span class="xref std std-ref">locally-disable-grad-doc</span> for a comparison between
    453. 

  453. <cite>.eval()</cite> and several similar mechanisms that may be confused with it.</p>
    454. 

  454. <dl class="field-list simple">
    455. 

  455. <dt class="field-odd">Returns</dt>
    456. 

  456. <dd class="field-odd"><p>self</p>
    457. 

  457. </dd>
    458. 

  458. <dt class="field-even">Return type</dt>
    459. 

  459. <dd class="field-even"><p>Module</p>
    460. 

  460. </dd>
    461. 

  461. </dl>
    462. 

  462. </dd></dl>
    463. 

  463. 

  464. <dl class="py method">
    465. 

  465. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.extra_repr">
    466. 

  466. <span class="sig-name descname"><span class="pre">extra_repr</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.extra_repr" title="Permalink to this definition">¶</a></dt>
    467. 

  467. <dd><p>Set the extra representation of the module</p>
    468. 

  468. <p>To print customized extra information, you should re-implement
    469. 

  469. this method in your own modules. Both single-line and multi-line
    470. 

  470. strings are acceptable.</p>
    471. 

  471. </dd></dl>
    472. 

  472. 

  473. <dl class="py method">
    474. 

  474. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.float">
    475. 

  475. <span class="sig-name descname"><span class="pre">float</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.float" title="Permalink to this definition">¶</a></dt>
    476. 

  476. <dd><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">float</span></code> datatype.</p>
    477. 

  477. <div class="admonition note">
    478. 

  478. <p class="admonition-title">Note</p>
    479. 

  479. <p>This method modifies the module in-place.</p>
    480. 

  480. </div>
    481. 

  481. <dl class="field-list simple">
    482. 

  482. <dt class="field-odd">Returns</dt>
    483. 

  483. <dd class="field-odd"><p>self</p>
    484. 

  484. </dd>
    485. 

  485. <dt class="field-even">Return type</dt>
    486. 

  486. <dd class="field-even"><p>Module</p>
    487. 

  487. </dd>
    488. 

  488. </dl>
    489. 

  489. </dd></dl>
    490. 

  490. 

  491. <dl class="py method">
    492. 

  492. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward">
    493. 

  493. <span class="sig-name descname"><span class="pre">forward</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">x</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">min_std</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">-</span> <span class="pre">20</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_std</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">2</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">MultivariateNormal</span></span></span><a class="reference internal" href="../_modules/draugr/torch_utilities/architectures/distributional/normal.html#MultiVariateNormalMLP.forward"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward" title="Permalink to this definition">¶</a></dt>
    494. 

  494. <dd><dl class="field-list simple">
    495. 

  495. <dt class="field-odd">Parameters</dt>
    496. 

  496. <dd class="field-odd"><ul class="simple">
    497. 

  497. <li><p><strong>x</strong> – </p></li>
    498. 

  498. <li><p><strong>min_std</strong> – </p></li>
    499. 

  499. <li><p><strong>max_std</strong> – </p></li>
    500. 

  500. <li><p><strong>kwargs</strong> – </p></li>
    501. 

  501. </ul>
    502. 

  502. </dd>
    503. 

  503. <dt class="field-even">Returns</dt>
    504. 

  504. <dd class="field-even"><p></p>
    505. 

  505. </dd>
    506. 

  506. <dt class="field-odd">Return type</dt>
    507. 

  507. <dd class="field-odd"><p></p>
    508. 

  508. </dd>
    509. 

  509. </dl>
    510. 

  510. </dd></dl>
    511. 

  511. 

  512. <dl class="py method">
    513. 

  513. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_buffer">
    514. 

  514. <span class="sig-name descname"><span class="pre">get_buffer</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">Tensor</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_buffer" title="Permalink to this definition">¶</a></dt>
    515. 

  515. <dd><p>Returns the buffer given by <code class="docutils literal notranslate"><span class="pre">target</span></code> if it exists,
    516. 

  516. otherwise throws an error.</p>
    517. 

  517. <p>See the docstring for <code class="docutils literal notranslate"><span class="pre">get_submodule</span></code> for a more detailed
    518. 

  518. explanation of this method’s functionality as well as how to
    519. 

  519. correctly specify <code class="docutils literal notranslate"><span class="pre">target</span></code>.</p>
    520. 

  520. <dl class="field-list simple">
    521. 

  521. <dt class="field-odd">Parameters</dt>
    522. 

  522. <dd class="field-odd"><p><strong>target</strong> – The fully-qualified string name of the buffer
    523. 

  523. to look for. (See <code class="docutils literal notranslate"><span class="pre">get_submodule</span></code> for how to specify a
    524. 

  524. fully-qualified string.)</p>
    525. 

  525. </dd>
    526. 

  526. <dt class="field-even">Returns</dt>
    527. 

  527. <dd class="field-even"><p>The buffer referenced by <code class="docutils literal notranslate"><span class="pre">target</span></code></p>
    528. 

  528. </dd>
    529. 

  529. <dt class="field-odd">Return type</dt>
    530. 

  530. <dd class="field-odd"><p>torch.Tensor</p>
    531. 

  531. </dd>
    532. 

  532. <dt class="field-even">Raises</dt>
    533. 

  533. <dd class="field-even"><p><a class="reference external" href="https://docs.python.org/3/library/exceptions.html#AttributeError" title="(in Python v3.10)"><strong>AttributeError</strong></a> – If the target string references an invalid
    534. 

  534.     path or resolves to something that is not a
    535. 

  535.     buffer</p>
    536. 

  536. </dd>
    537. 

  537. </dl>
    538. 

  538. </dd></dl>
    539. 

  539. 

  540. <dl class="py method">
    541. 

  541. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_extra_state">
    542. 

  542. <span class="sig-name descname"><span class="pre">get_extra_state</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Any" title="(in Python v3.10)"><span class="pre">Any</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_extra_state" title="Permalink to this definition">¶</a></dt>
    543. 

  543. <dd><p>Returns any extra state to include in the module’s state_dict.
    544. 

  544. Implement this and a corresponding <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.set_extra_state" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.set_extra_state"><code class="xref py py-func docutils literal notranslate"><span class="pre">set_extra_state()</span></code></a> for your module
    545. 

  545. if you need to store extra state. This function is called when building the
    546. 

  546. module’s <cite>state_dict()</cite>.</p>
    547. 

  547. <p>Note that extra state should be pickleable to ensure working serialization
    548. 

  548. of the state_dict. We only provide provide backwards compatibility guarantees
    549. 

  549. for serializing Tensors; other objects may break backwards compatibility if
    550. 

  550. their serialized pickled form changes.</p>
    551. 

  551. <dl class="field-list simple">
    552. 

  552. <dt class="field-odd">Returns</dt>
    553. 

  553. <dd class="field-odd"><p>Any extra state to store in the module’s state_dict</p>
    554. 

  554. </dd>
    555. 

  555. <dt class="field-even">Return type</dt>
    556. 

  556. <dd class="field-even"><p><a class="reference external" href="https://docs.python.org/3/library/functions.html#object" title="(in Python v3.10)">object</a></p>
    557. 

  557. </dd>
    558. 

  558. </dl>
    559. 

  559. </dd></dl>
    560. 

  560. 

  561. <dl class="py method">
    562. 

  562. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_parameter">
    563. 

  563. <span class="sig-name descname"><span class="pre">get_parameter</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">Parameter</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_parameter" title="Permalink to this definition">¶</a></dt>
    564. 

  564. <dd><p>Returns the parameter given by <code class="docutils literal notranslate"><span class="pre">target</span></code> if it exists,
    565. 

  565. otherwise throws an error.</p>
    566. 

  566. <p>See the docstring for <code class="docutils literal notranslate"><span class="pre">get_submodule</span></code> for a more detailed
    567. 

  567. explanation of this method’s functionality as well as how to
    568. 

  568. correctly specify <code class="docutils literal notranslate"><span class="pre">target</span></code>.</p>
    569. 

  569. <dl class="field-list simple">
    570. 

  570. <dt class="field-odd">Parameters</dt>
    571. 

  571. <dd class="field-odd"><p><strong>target</strong> – The fully-qualified string name of the Parameter
    572. 

  572. to look for. (See <code class="docutils literal notranslate"><span class="pre">get_submodule</span></code> for how to specify a
    573. 

  573. fully-qualified string.)</p>
    574. 

  574. </dd>
    575. 

  575. <dt class="field-even">Returns</dt>
    576. 

  576. <dd class="field-even"><p>The Parameter referenced by <code class="docutils literal notranslate"><span class="pre">target</span></code></p>
    577. 

  577. </dd>
    578. 

  578. <dt class="field-odd">Return type</dt>
    579. 

  579. <dd class="field-odd"><p>torch.nn.Parameter</p>
    580. 

  580. </dd>
    581. 

  581. <dt class="field-even">Raises</dt>
    582. 

  582. <dd class="field-even"><p><a class="reference external" href="https://docs.python.org/3/library/exceptions.html#AttributeError" title="(in Python v3.10)"><strong>AttributeError</strong></a> – If the target string references an invalid
    583. 

  583.     path or resolves to something that is not an
    584. 

  584.     <code class="docutils literal notranslate"><span class="pre">nn.Parameter</span></code></p>
    585. 

  585. </dd>
    586. 

  586. </dl>
    587. 

  587. </dd></dl>
    588. 

  588. 

  589. <dl class="py method">
    590. 

  590. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_submodule">
    591. 

  591. <span class="sig-name descname"><span class="pre">get_submodule</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">Module</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_submodule" title="Permalink to this definition">¶</a></dt>
    592. 

  592. <dd><p>Returns the submodule given by <code class="docutils literal notranslate"><span class="pre">target</span></code> if it exists,
    593. 

  593. otherwise throws an error.</p>
    594. 

  594. <p>For example, let’s say you have an <code class="docutils literal notranslate"><span class="pre">nn.Module</span></code> <code class="docutils literal notranslate"><span class="pre">A</span></code> that
    595. 

  595. looks like this:</p>
    596. 

  596. <div class="highlight-text notranslate"><div class="highlight"><pre><span></span>A(
    597. 

  597.     (net_b): Module(
    598. 

  598.         (net_c): Module(
    599. 

  599.             (conv): Conv2d(16, 33, kernel_size=(3, 3), stride=(2, 2))
    600. 

  600.         )
    601. 

  601.         (linear): Linear(in_features=100, out_features=200, bias=True)
    602. 

  602.     )
    603. 

  603. )
    604. 

  604. </pre></div>
    605. 

  605. </div>
    606. 

  606. <p>(The diagram shows an <code class="docutils literal notranslate"><span class="pre">nn.Module</span></code> <code class="docutils literal notranslate"><span class="pre">A</span></code>. <code class="docutils literal notranslate"><span class="pre">A</span></code> has a nested
    607. 

  607. submodule <code class="docutils literal notranslate"><span class="pre">net_b</span></code>, which itself has two submodules <code class="docutils literal notranslate"><span class="pre">net_c</span></code>
    608. 

  608. and <code class="docutils literal notranslate"><span class="pre">linear</span></code>. <code class="docutils literal notranslate"><span class="pre">net_c</span></code> then has a submodule <code class="docutils literal notranslate"><span class="pre">conv</span></code>.)</p>
    609. 

  609. <p>To check whether or not we have the <code class="docutils literal notranslate"><span class="pre">linear</span></code> submodule, we
    610. 

  610. would call <code class="docutils literal notranslate"><span class="pre">get_submodule(&quot;net_b.linear&quot;)</span></code>. To check whether
    611. 

  611. we have the <code class="docutils literal notranslate"><span class="pre">conv</span></code> submodule, we would call
    612. 

  612. <code class="docutils literal notranslate"><span class="pre">get_submodule(&quot;net_b.net_c.conv&quot;)</span></code>.</p>
    613. 

  613. <p>The runtime of <code class="docutils literal notranslate"><span class="pre">get_submodule</span></code> is bounded by the degree
    614. 

  614. of module nesting in <code class="docutils literal notranslate"><span class="pre">target</span></code>. A query against
    615. 

  615. <code class="docutils literal notranslate"><span class="pre">named_modules</span></code> achieves the same result, but it is O(N) in
    616. 

  616. the number of transitive modules. So, for a simple check to see
    617. 

  617. if some submodule exists, <code class="docutils literal notranslate"><span class="pre">get_submodule</span></code> should always be
    618. 

  618. used.</p>
    619. 

  619. <dl class="field-list simple">
    620. 

  620. <dt class="field-odd">Parameters</dt>
    621. 

  621. <dd class="field-odd"><p><strong>target</strong> – The fully-qualified string name of the submodule
    622. 

  622. to look for. (See above example for how to specify a
    623. 

  623. fully-qualified string.)</p>
    624. 

  624. </dd>
    625. 

  625. <dt class="field-even">Returns</dt>
    626. 

  626. <dd class="field-even"><p>The submodule referenced by <code class="docutils literal notranslate"><span class="pre">target</span></code></p>
    627. 

  627. </dd>
    628. 

  628. <dt class="field-odd">Return type</dt>
    629. 

  629. <dd class="field-odd"><p>torch.nn.Module</p>
    630. 

  630. </dd>
    631. 

  631. <dt class="field-even">Raises</dt>
    632. 

  632. <dd class="field-even"><p><a class="reference external" href="https://docs.python.org/3/library/exceptions.html#AttributeError" title="(in Python v3.10)"><strong>AttributeError</strong></a> – If the target string references an invalid
    633. 

  633.     path or resolves to something that is not an
    634. 

  634.     <code class="docutils literal notranslate"><span class="pre">nn.Module</span></code></p>
    635. 

  635. </dd>
    636. 

  636. </dl>
    637. 

  637. </dd></dl>
    638. 

  638. 

  639. <dl class="py method">
    640. 

  640. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.half">
    641. 

  641. <span class="sig-name descname"><span class="pre">half</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.half" title="Permalink to this definition">¶</a></dt>
    642. 

  642. <dd><p>Casts all floating point parameters and buffers to <code class="docutils literal notranslate"><span class="pre">half</span></code> datatype.</p>
    643. 

  643. <div class="admonition note">
    644. 

  644. <p class="admonition-title">Note</p>
    645. 

  645. <p>This method modifies the module in-place.</p>
    646. 

  646. </div>
    647. 

  647. <dl class="field-list simple">
    648. 

  648. <dt class="field-odd">Returns</dt>
    649. 

  649. <dd class="field-odd"><p>self</p>
    650. 

  650. </dd>
    651. 

  651. <dt class="field-even">Return type</dt>
    652. 

  652. <dd class="field-even"><p>Module</p>
    653. 

  653. </dd>
    654. 

  654. </dl>
    655. 

  655. </dd></dl>
    656. 

  656. 

  657. <dl class="py method">
    658. 

  658. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_input_shape">
    659. 

  659. <em class="property"><span class="pre">static</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">infer_input_shape</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">input_shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Sequence" title="(in Python v3.10)"><span class="pre">Sequence</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Sequence" title="(in Python v3.10)"><span class="pre">Sequence</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_input_shape" title="Permalink to this definition">¶</a></dt>
    660. 

  660. <dd><blockquote>
    661. 

  661. <div><dl class="field-list simple">
    662. 

  662. <dt class="field-odd">return</dt>
    663. 

  663. <dd class="field-odd"><p></p></dd>
    664. 

  664. <dt class="field-even">rtype</dt>
    665. 

  665. <dd class="field-even"><p></p></dd>
    666. 

  666. </dl>
    667. 

  667. </div></blockquote>
    668. 

  668. <dl class="field-list simple">
    669. 

  669. <dt class="field-odd">Parameters</dt>
    670. 

  670. <dd class="field-odd"><p><strong>input_shape</strong> – </p>
    671. 

  671. </dd>
    672. 

  672. <dt class="field-even">Returns</dt>
    673. 

  673. <dd class="field-even"><p></p>
    674. 

  674. </dd>
    675. 

  675. </dl>
    676. 

  676. </dd></dl>
    677. 

  677. 

  678. <dl class="py method">
    679. 

  679. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_output_shape">
    680. 

  680. <em class="property"><span class="pre">static</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">infer_output_shape</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">output_shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Sequence" title="(in Python v3.10)"><span class="pre">Sequence</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Sequence" title="(in Python v3.10)"><span class="pre">Sequence</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.infer_output_shape" title="Permalink to this definition">¶</a></dt>
    681. 

  681. <dd><blockquote>
    682. 

  682. <div><dl class="field-list simple">
    683. 

  683. <dt class="field-odd">return</dt>
    684. 

  684. <dd class="field-odd"><p></p></dd>
    685. 

  685. <dt class="field-even">rtype</dt>
    686. 

  686. <dd class="field-even"><p></p></dd>
    687. 

  687. </dl>
    688. 

  688. </div></blockquote>
    689. 

  689. <dl class="field-list simple">
    690. 

  690. <dt class="field-odd">Parameters</dt>
    691. 

  691. <dd class="field-odd"><p><strong>output_shape</strong> – </p>
    692. 

  692. </dd>
    693. 

  693. <dt class="field-even">Returns</dt>
    694. 

  694. <dd class="field-even"><p></p>
    695. 

  695. </dd>
    696. 

  696. </dl>
    697. 

  697. </dd></dl>
    698. 

  698. 

  699. <dl class="py property">
    700. 

  700. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.input_shape">
    701. 

  701. <em class="property"><span class="pre">property</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">input_shape</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Sequence" title="(in Python v3.10)"><span class="pre">Sequence</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.input_shape" title="Permalink to this definition">¶</a></dt>
    702. 

  702. <dd><p>return:
    703. 

  703. :rtype:</p>
    704. 

  704. </dd></dl>
    705. 

  705. 

  706. <dl class="py method">
    707. 

  707. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.ipu">
    708. 

  708. <span class="sig-name descname"><span class="pre">ipu</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">device</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.ipu" title="Permalink to this definition">¶</a></dt>
    709. 

  709. <dd><p>Moves all model parameters and buffers to the IPU.</p>
    710. 

  710. <p>This also makes associated parameters and buffers different objects. So
    711. 

  711. it should be called before constructing optimizer if the module will
    712. 

  712. live on IPU while being optimized.</p>
    713. 

  713. <div class="admonition note">
    714. 

  714. <p class="admonition-title">Note</p>
    715. 

  715. <p>This method modifies the module in-place.</p>
    716. 

  716. </div>
    717. 

  717. <dl class="field-list simple">
    718. 

  718. <dt class="field-odd">Parameters</dt>
    719. 

  719. <dd class="field-odd"><p><strong>device</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><em>int</em></a><em>, </em><em>optional</em>) – if specified, all parameters will be
    720. 

  720. copied to that device</p>
    721. 

  721. </dd>
    722. 

  722. <dt class="field-even">Returns</dt>
    723. 

  723. <dd class="field-even"><p>self</p>
    724. 

  724. </dd>
    725. 

  725. <dt class="field-odd">Return type</dt>
    726. 

  726. <dd class="field-odd"><p>Module</p>
    727. 

  727. </dd>
    728. 

  728. </dl>
    729. 

  729. </dd></dl>
    730. 

  730. 

  731. <dl class="py method">
    732. 

  732. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict">
    733. 

  733. <span class="sig-name descname"><span class="pre">load_state_dict</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">state_dict</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Mapping" title="(in Python v3.10)"><span class="pre">Mapping</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Any" title="(in Python v3.10)"><span class="pre">Any</span></a><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">strict</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict" title="Permalink to this definition">¶</a></dt>
    734. 

  734. <dd><p>Copies parameters and buffers from <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a> into
    735. 

  735. this module and its descendants. If <code class="xref py py-attr docutils literal notranslate"><span class="pre">strict</span></code> is <code class="docutils literal notranslate"><span class="pre">True</span></code>, then
    736. 

  736. the keys of <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a> must exactly match the keys returned
    737. 

  737. by this module’s <code class="xref py py-meth docutils literal notranslate"><span class="pre">state_dict()</span></code> function.</p>
    738. 

  738. <dl class="field-list simple">
    739. 

  739. <dt class="field-odd">Parameters</dt>
    740. 

  740. <dd class="field-odd"><ul class="simple">
    741. 

  741. <li><p><strong>state_dict</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#dict" title="(in Python v3.10)"><em>dict</em></a>) – a dict containing parameters and
    742. 

  742. persistent buffers.</p></li>
    743. 

  743. <li><p><strong>strict</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a><em>, </em><em>optional</em>) – whether to strictly enforce that the keys
    744. 

  744. in <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a> match the keys returned by this module’s
    745. 

  745. <code class="xref py py-meth docutils literal notranslate"><span class="pre">state_dict()</span></code> function. Default: <code class="docutils literal notranslate"><span class="pre">True</span></code></p></li>
    746. 

  746. </ul>
    747. 

  747. </dd>
    748. 

  748. <dt class="field-even">Returns</dt>
    749. 

  749. <dd class="field-even"><p><ul class="simple">
    750. 

  750. <li><p><strong>missing_keys</strong> is a list of str containing the missing keys</p></li>
    751. 

  751. <li><p><strong>unexpected_keys</strong> is a list of str containing the unexpected keys</p></li>
    752. 

  752. </ul>
    753. 

  753. </p>
    754. 

  754. </dd>
    755. 

  755. <dt class="field-odd">Return type</dt>
    756. 

  756. <dd class="field-odd"><p><code class="docutils literal notranslate"><span class="pre">NamedTuple</span></code> with <code class="docutils literal notranslate"><span class="pre">missing_keys</span></code> and <code class="docutils literal notranslate"><span class="pre">unexpected_keys</span></code> fields</p>
    757. 

  757. </dd>
    758. 

  758. </dl>
    759. 

  759. <div class="admonition note">
    760. 

  760. <p class="admonition-title">Note</p>
    761. 

  761. <p>If a parameter or buffer is registered as <code class="docutils literal notranslate"><span class="pre">None</span></code> and its corresponding key
    762. 

  762. exists in <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a>, <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict"><code class="xref py py-meth docutils literal notranslate"><span class="pre">load_state_dict()</span></code></a> will raise a
    763. 

  763. <code class="docutils literal notranslate"><span class="pre">RuntimeError</span></code>.</p>
    764. 

  764. </div>
    765. 

  765. </dd></dl>
    766. 

  766. 

  767. <dl class="py method">
    768. 

  768. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.modules">
    769. 

  769. <span class="sig-name descname"><span class="pre">modules</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Iterator" title="(in Python v3.10)"><span class="pre">Iterator</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.modules" title="Permalink to this definition">¶</a></dt>
    770. 

  770. <dd><p>Returns an iterator over all modules in the network.</p>
    771. 

  771. <dl class="field-list simple">
    772. 

  772. <dt class="field-odd">Yields</dt>
    773. 

  773. <dd class="field-odd"><p><em>Module</em> – a module in the network</p>
    774. 

  774. </dd>
    775. 

  775. </dl>
    776. 

  776. <div class="admonition note">
    777. 

  777. <p class="admonition-title">Note</p>
    778. 

  778. <p>Duplicate modules are returned only once. In the following
    779. 

  779. example, <code class="docutils literal notranslate"><span class="pre">l</span></code> will be returned only once.</p>
    780. 

  780. </div>
    781. 

  781. <p>Example:</p>
    782. 

  782. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="n">l</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span>
    783. 

  783. <span class="gp">&gt;&gt;&gt; </span><span class="n">net</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Sequential</span><span class="p">(</span><span class="n">l</span><span class="p">,</span> <span class="n">l</span><span class="p">)</span>
    784. 

  784. <span class="gp">&gt;&gt;&gt; </span><span class="k">for</span> <span class="n">idx</span><span class="p">,</span> <span class="n">m</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">net</span><span class="o">.</span><span class="n">modules</span><span class="p">()):</span>
    785. 

  785. <span class="go">        print(idx, &#39;-&gt;&#39;, m)</span>
    786. 

  786. 

  787. <span class="go">0 -&gt; Sequential(</span>
    788. 

  788. <span class="go">  (0): Linear(in_features=2, out_features=2, bias=True)</span>
    789. 

  789. <span class="go">  (1): Linear(in_features=2, out_features=2, bias=True)</span>
    790. 

  790. <span class="go">)</span>
    791. 

  791. <span class="go">1 -&gt; Linear(in_features=2, out_features=2, bias=True)</span>
    792. 

  792. </pre></div>
    793. 

  793. </div>
    794. 

  794. </dd></dl>
    795. 

  795. 

  796. <dl class="py method">
    797. 

  797. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_buffers">
    798. 

  798. <span class="sig-name descname"><span class="pre">named_buffers</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">prefix</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">''</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">recurse</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Iterator" title="(in Python v3.10)"><span class="pre">Iterator</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.10)"><span class="pre">Tuple</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_buffers" title="Permalink to this definition">¶</a></dt>
    799. 

  799. <dd><p>Returns an iterator over module buffers, yielding both the
    800. 

  800. name of the buffer as well as the buffer itself.</p>
    801. 

  801. <dl class="field-list simple">
    802. 

  802. <dt class="field-odd">Parameters</dt>
    803. 

  803. <dd class="field-odd"><ul class="simple">
    804. 

  804. <li><p><strong>prefix</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><em>str</em></a>) – prefix to prepend to all buffer names.</p></li>
    805. 

  805. <li><p><strong>recurse</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – if True, then yields buffers of this module
    806. 

  806. and all submodules. Otherwise, yields only buffers that
    807. 

  807. are direct members of this module.</p></li>
    808. 

  808. </ul>
    809. 

  809. </dd>
    810. 

  810. <dt class="field-even">Yields</dt>
    811. 

  811. <dd class="field-even"><p><em>(string, torch.Tensor)</em> – Tuple containing the name and buffer</p>
    812. 

  812. </dd>
    813. 

  813. </dl>
    814. 

  814. <p>Example:</p>
    815. 

  815. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">buf</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">named_buffers</span><span class="p">():</span>
    816. 

  816. <span class="gp">&gt;&gt;&gt; </span>   <span class="k">if</span> <span class="n">name</span> <span class="ow">in</span> <span class="p">[</span><span class="s1">&#39;running_var&#39;</span><span class="p">]:</span>
    817. 

  817. <span class="gp">&gt;&gt;&gt; </span>       <span class="nb">print</span><span class="p">(</span><span class="n">buf</span><span class="o">.</span><span class="n">size</span><span class="p">())</span>
    818. 

  818. </pre></div>
    819. 

  819. </div>
    820. 

  820. </dd></dl>
    821. 

  821. 

  822. <dl class="py method">
    823. 

  823. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_children">
    824. 

  824. <span class="sig-name descname"><span class="pre">named_children</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Iterator" title="(in Python v3.10)"><span class="pre">Iterator</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.10)"><span class="pre">Tuple</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_children" title="Permalink to this definition">¶</a></dt>
    825. 

  825. <dd><p>Returns an iterator over immediate children modules, yielding both
    826. 

  826. the name of the module as well as the module itself.</p>
    827. 

  827. <dl class="field-list simple">
    828. 

  828. <dt class="field-odd">Yields</dt>
    829. 

  829. <dd class="field-odd"><p><em>(string, Module)</em> – Tuple containing a name and child module</p>
    830. 

  830. </dd>
    831. 

  831. </dl>
    832. 

  832. <p>Example:</p>
    833. 

  833. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">module</span> <span class="ow">in</span> <span class="n">model</span><span class="o">.</span><span class="n">named_children</span><span class="p">():</span>
    834. 

  834. <span class="gp">&gt;&gt;&gt; </span>    <span class="k">if</span> <span class="n">name</span> <span class="ow">in</span> <span class="p">[</span><span class="s1">&#39;conv4&#39;</span><span class="p">,</span> <span class="s1">&#39;conv5&#39;</span><span class="p">]:</span>
    835. 

  835. <span class="gp">&gt;&gt;&gt; </span>        <span class="nb">print</span><span class="p">(</span><span class="n">module</span><span class="p">)</span>
    836. 

  836. </pre></div>
    837. 

  837. </div>
    838. 

  838. </dd></dl>
    839. 

  839. 

  840. <dl class="py method">
    841. 

  841. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_modules">
    842. 

  842. <span class="sig-name descname"><span class="pre">named_modules</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">memo</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Set" title="(in Python v3.10)"><span class="pre">Set</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">prefix</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">''</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">remove_duplicate</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_modules" title="Permalink to this definition">¶</a></dt>
    843. 

  843. <dd><p>Returns an iterator over all modules in the network, yielding
    844. 

  844. both the name of the module as well as the module itself.</p>
    845. 

  845. <dl class="field-list simple">
    846. 

  846. <dt class="field-odd">Parameters</dt>
    847. 

  847. <dd class="field-odd"><ul class="simple">
    848. 

  848. <li><p><strong>memo</strong> – a memo to store the set of modules already added to the result</p></li>
    849. 

  849. <li><p><strong>prefix</strong> – a prefix that will be added to the name of the module</p></li>
    850. 

  850. <li><p><strong>remove_duplicate</strong> – whether to remove the duplicated module instances in the result
    851. 

  851. or not</p></li>
    852. 

  852. </ul>
    853. 

  853. </dd>
    854. 

  854. <dt class="field-even">Yields</dt>
    855. 

  855. <dd class="field-even"><p><em>(string, Module)</em> – Tuple of name and module</p>
    856. 

  856. </dd>
    857. 

  857. </dl>
    858. 

  858. <div class="admonition note">
    859. 

  859. <p class="admonition-title">Note</p>
    860. 

  860. <p>Duplicate modules are returned only once. In the following
    861. 

  861. example, <code class="docutils literal notranslate"><span class="pre">l</span></code> will be returned only once.</p>
    862. 

  862. </div>
    863. 

  863. <p>Example:</p>
    864. 

  864. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="n">l</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span>
    865. 

  865. <span class="gp">&gt;&gt;&gt; </span><span class="n">net</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Sequential</span><span class="p">(</span><span class="n">l</span><span class="p">,</span> <span class="n">l</span><span class="p">)</span>
    866. 

  866. <span class="gp">&gt;&gt;&gt; </span><span class="k">for</span> <span class="n">idx</span><span class="p">,</span> <span class="n">m</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">net</span><span class="o">.</span><span class="n">named_modules</span><span class="p">()):</span>
    867. 

  867. <span class="go">        print(idx, &#39;-&gt;&#39;, m)</span>
    868. 

  868. 

  869. <span class="go">0 -&gt; (&#39;&#39;, Sequential(</span>
    870. 

  870. <span class="go">  (0): Linear(in_features=2, out_features=2, bias=True)</span>
    871. 

  871. <span class="go">  (1): Linear(in_features=2, out_features=2, bias=True)</span>
    872. 

  872. <span class="go">))</span>
    873. 

  873. <span class="go">1 -&gt; (&#39;0&#39;, Linear(in_features=2, out_features=2, bias=True))</span>
    874. 

  874. </pre></div>
    875. 

  875. </div>
    876. 

  876. </dd></dl>
    877. 

  877. 

  878. <dl class="py method">
    879. 

  879. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_parameters">
    880. 

  880. <span class="sig-name descname"><span class="pre">named_parameters</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">prefix</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">''</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">recurse</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Iterator" title="(in Python v3.10)"><span class="pre">Iterator</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.10)"><span class="pre">Tuple</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Parameter</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.named_parameters" title="Permalink to this definition">¶</a></dt>
    881. 

  881. <dd><p>Returns an iterator over module parameters, yielding both the
    882. 

  882. name of the parameter as well as the parameter itself.</p>
    883. 

  883. <dl class="field-list simple">
    884. 

  884. <dt class="field-odd">Parameters</dt>
    885. 

  885. <dd class="field-odd"><ul class="simple">
    886. 

  886. <li><p><strong>prefix</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><em>str</em></a>) – prefix to prepend to all parameter names.</p></li>
    887. 

  887. <li><p><strong>recurse</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – if True, then yields parameters of this module
    888. 

  888. and all submodules. Otherwise, yields only parameters that
    889. 

  889. are direct members of this module.</p></li>
    890. 

  890. </ul>
    891. 

  891. </dd>
    892. 

  892. <dt class="field-even">Yields</dt>
    893. 

  893. <dd class="field-even"><p><em>(string, Parameter)</em> – Tuple containing the name and parameter</p>
    894. 

  894. </dd>
    895. 

  895. </dl>
    896. 

  896. <p>Example:</p>
    897. 

  897. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">param</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">named_parameters</span><span class="p">():</span>
    898. 

  898. <span class="gp">&gt;&gt;&gt; </span>   <span class="k">if</span> <span class="n">name</span> <span class="ow">in</span> <span class="p">[</span><span class="s1">&#39;bias&#39;</span><span class="p">]:</span>
    899. 

  899. <span class="gp">&gt;&gt;&gt; </span>       <span class="nb">print</span><span class="p">(</span><span class="n">param</span><span class="o">.</span><span class="n">size</span><span class="p">())</span>
    900. 

  900. </pre></div>
    901. 

  901. </div>
    902. 

  902. </dd></dl>
    903. 

  903. 

  904. <dl class="py property">
    905. 

  905. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.output_shape">
    906. 

  906. <em class="property"><span class="pre">property</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">output_shape</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Sequence" title="(in Python v3.10)"><span class="pre">Sequence</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.output_shape" title="Permalink to this definition">¶</a></dt>
    907. 

  907. <dd><p>return:
    908. 

  908. :rtype:</p>
    909. 

  909. </dd></dl>
    910. 

  910. 

  911. <dl class="py method">
    912. 

  912. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.parameters">
    913. 

  913. <span class="sig-name descname"><span class="pre">parameters</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">recurse</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Iterator" title="(in Python v3.10)"><span class="pre">Iterator</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Parameter</span><span class="p"><span class="pre">]</span></span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.parameters" title="Permalink to this definition">¶</a></dt>
    914. 

  914. <dd><p>Returns an iterator over module parameters.</p>
    915. 

  915. <p>This is typically passed to an optimizer.</p>
    916. 

  916. <dl class="field-list simple">
    917. 

  917. <dt class="field-odd">Parameters</dt>
    918. 

  918. <dd class="field-odd"><p><strong>recurse</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – if True, then yields parameters of this module
    919. 

  919. and all submodules. Otherwise, yields only parameters that
    920. 

  920. are direct members of this module.</p>
    921. 

  921. </dd>
    922. 

  922. <dt class="field-even">Yields</dt>
    923. 

  923. <dd class="field-even"><p><em>Parameter</em> – module parameter</p>
    924. 

  924. </dd>
    925. 

  925. </dl>
    926. 

  926. <p>Example:</p>
    927. 

  927. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="k">for</span> <span class="n">param</span> <span class="ow">in</span> <span class="n">model</span><span class="o">.</span><span class="n">parameters</span><span class="p">():</span>
    928. 

  928. <span class="gp">&gt;&gt;&gt; </span>    <span class="nb">print</span><span class="p">(</span><span class="nb">type</span><span class="p">(</span><span class="n">param</span><span class="p">),</span> <span class="n">param</span><span class="o">.</span><span class="n">size</span><span class="p">())</span>
    929. 

  929. <span class="go">&lt;class &#39;torch.Tensor&#39;&gt; (20L,)</span>
    930. 

  930. <span class="go">&lt;class &#39;torch.Tensor&#39;&gt; (20L, 1L, 5L, 5L)</span>
    931. 

  931. </pre></div>
    932. 

  932. </div>
    933. 

  933. </dd></dl>
    934. 

  934. 

  935. <dl class="py method">
    936. 

  936. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_backward_hook">
    937. 

  937. <span class="sig-name descname"><span class="pre">register_backward_hook</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">hook</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Callable" title="(in Python v3.10)"><span class="pre">Callable</span></a><span class="p"><span class="pre">[</span></span><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.10)"><span class="pre">Tuple</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Tensor</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.10)"><span class="pre">Tuple</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Tensor</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">RemovableHandle</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_backward_hook" title="Permalink to this definition">¶</a></dt>
    938. 

  938. <dd><p>Registers a backward hook on the module.</p>
    939. 

  939. <p>This function is deprecated in favor of <code class="xref py py-meth docutils literal notranslate"><span class="pre">register_full_backward_hook()</span></code> and
    940. 

  940. the behavior of this function will change in future versions.</p>
    941. 

  941. <dl class="field-list simple">
    942. 

  942. <dt class="field-odd">Returns</dt>
    943. 

  943. <dd class="field-odd"><p>a handle that can be used to remove the added hook by calling
    944. 

  944. <code class="docutils literal notranslate"><span class="pre">handle.remove()</span></code></p>
    945. 

  945. </dd>
    946. 

  946. <dt class="field-even">Return type</dt>
    947. 

  947. <dd class="field-even"><p><code class="xref py py-class docutils literal notranslate"><span class="pre">torch.utils.hooks.RemovableHandle</span></code></p>
    948. 

  948. </dd>
    949. 

  949. </dl>
    950. 

  950. </dd></dl>
    951. 

  951. 

  952. <dl class="py method">
    953. 

  953. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_buffer">
    954. 

  954. <span class="sig-name descname"><span class="pre">register_buffer</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensor</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">persistent</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_buffer" title="Permalink to this definition">¶</a></dt>
    955. 

  955. <dd><p>Adds a buffer to the module.</p>
    956. 

  956. <p>This is typically used to register a buffer that should not to be
    957. 

  957. considered a model parameter. For example, BatchNorm’s <code class="docutils literal notranslate"><span class="pre">running_mean</span></code>
    958. 

  958. is not a parameter, but is part of the module’s state. Buffers, by
    959. 

  959. default, are persistent and will be saved alongside parameters. This
    960. 

  960. behavior can be changed by setting <code class="xref py py-attr docutils literal notranslate"><span class="pre">persistent</span></code> to <code class="docutils literal notranslate"><span class="pre">False</span></code>. The
    961. 

  961. only difference between a persistent buffer and a non-persistent buffer
    962. 

  962. is that the latter will not be a part of this module’s
    963. 

  963. <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a>.</p>
    964. 

  964. <p>Buffers can be accessed as attributes using given names.</p>
    965. 

  965. <dl class="field-list simple">
    966. 

  966. <dt class="field-odd">Parameters</dt>
    967. 

  967. <dd class="field-odd"><ul class="simple">
    968. 

  968. <li><p><strong>name</strong> (<em>string</em>) – name of the buffer. The buffer can be accessed
    969. 

  969. from this module using the given name</p></li>
    970. 

  970. <li><p><strong>tensor</strong> (<em>Tensor</em><em> or </em><em>None</em>) – buffer to be registered. If <code class="docutils literal notranslate"><span class="pre">None</span></code>, then operations
    971. 

  971. that run on buffers, such as <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda"><code class="xref py py-attr docutils literal notranslate"><span class="pre">cuda</span></code></a>, are ignored. If <code class="docutils literal notranslate"><span class="pre">None</span></code>,
    972. 

  972. the buffer is <strong>not</strong> included in the module’s <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a>.</p></li>
    973. 

  973. <li><p><strong>persistent</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – whether the buffer is part of this module’s
    974. 

  974. <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a>.</p></li>
    975. 

  975. </ul>
    976. 

  976. </dd>
    977. 

  977. </dl>
    978. 

  978. <p>Example:</p>
    979. 

  979. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="bp">self</span><span class="o">.</span><span class="n">register_buffer</span><span class="p">(</span><span class="s1">&#39;running_mean&#39;</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">zeros</span><span class="p">(</span><span class="n">num_features</span><span class="p">))</span>
    980. 

  980. </pre></div>
    981. 

  981. </div>
    982. 

  982. </dd></dl>
    983. 

  983. 

  984. <dl class="py method">
    985. 

  985. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_hook">
    986. 

  986. <span class="sig-name descname"><span class="pre">register_forward_hook</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">hook</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Callable" title="(in Python v3.10)"><span class="pre">Callable</span></a><span class="p"><span class="pre">[</span></span><span class="p"><span class="pre">[</span></span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">RemovableHandle</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_hook" title="Permalink to this definition">¶</a></dt>
    987. 

  987. <dd><p>Registers a forward hook on the module.</p>
    988. 

  988. <p>The hook will be called every time after <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward"><code class="xref py py-func docutils literal notranslate"><span class="pre">forward()</span></code></a> has computed an output.
    989. 

  989. It should have the following signature:</p>
    990. 

  990. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="n">hook</span><span class="p">(</span><span class="n">module</span><span class="p">,</span> <span class="nb">input</span><span class="p">,</span> <span class="n">output</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span> <span class="ow">or</span> <span class="n">modified</span> <span class="n">output</span>
    991. 

  991. </pre></div>
    992. 

  992. </div>
    993. 

  993. <p>The input contains only the positional arguments given to the module.
    994. 

  994. Keyword arguments won’t be passed to the hooks and only to the <code class="docutils literal notranslate"><span class="pre">forward</span></code>.
    995. 

  995. The hook can modify the output. It can modify the input inplace but
    996. 

  996. it will not have effect on forward since this is called after
    997. 

  997. <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward"><code class="xref py py-func docutils literal notranslate"><span class="pre">forward()</span></code></a> is called.</p>
    998. 

  998. <dl class="field-list simple">
    999. 

  999. <dt class="field-odd">Returns</dt>
    1000. 

  1000. <dd class="field-odd"><p>a handle that can be used to remove the added hook by calling
    1001. 

  1001. <code class="docutils literal notranslate"><span class="pre">handle.remove()</span></code></p>
    1002. 

  1002. </dd>
    1003. 

  1003. <dt class="field-even">Return type</dt>
    1004. 

  1004. <dd class="field-even"><p><code class="xref py py-class docutils literal notranslate"><span class="pre">torch.utils.hooks.RemovableHandle</span></code></p>
    1005. 

  1005. </dd>
    1006. 

  1006. </dl>
    1007. 

  1007. </dd></dl>
    1008. 

  1008. 

  1009. <dl class="py method">
    1010. 

  1010. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_pre_hook">
    1011. 

  1011. <span class="sig-name descname"><span class="pre">register_forward_pre_hook</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">hook</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Callable" title="(in Python v3.10)"><span class="pre">Callable</span></a><span class="p"><span class="pre">[</span></span><span class="p"><span class="pre">[</span></span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">RemovableHandle</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_forward_pre_hook" title="Permalink to this definition">¶</a></dt>
    1012. 

  1012. <dd><p>Registers a forward pre-hook on the module.</p>
    1013. 

  1013. <p>The hook will be called every time before <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.forward"><code class="xref py py-func docutils literal notranslate"><span class="pre">forward()</span></code></a> is invoked.
    1014. 

  1014. It should have the following signature:</p>
    1015. 

  1015. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="n">hook</span><span class="p">(</span><span class="n">module</span><span class="p">,</span> <span class="nb">input</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span> <span class="ow">or</span> <span class="n">modified</span> <span class="nb">input</span>
    1016. 

  1016. </pre></div>
    1017. 

  1017. </div>
    1018. 

  1018. <p>The input contains only the positional arguments given to the module.
    1019. 

  1019. Keyword arguments won’t be passed to the hooks and only to the <code class="docutils literal notranslate"><span class="pre">forward</span></code>.
    1020. 

  1020. The hook can modify the input. User can either return a tuple or a
    1021. 

  1021. single modified value in the hook. We will wrap the value into a tuple
    1022. 

  1022. if a single value is returned(unless that value is already a tuple).</p>
    1023. 

  1023. <dl class="field-list simple">
    1024. 

  1024. <dt class="field-odd">Returns</dt>
    1025. 

  1025. <dd class="field-odd"><p>a handle that can be used to remove the added hook by calling
    1026. 

  1026. <code class="docutils literal notranslate"><span class="pre">handle.remove()</span></code></p>
    1027. 

  1027. </dd>
    1028. 

  1028. <dt class="field-even">Return type</dt>
    1029. 

  1029. <dd class="field-even"><p><code class="xref py py-class docutils literal notranslate"><span class="pre">torch.utils.hooks.RemovableHandle</span></code></p>
    1030. 

  1030. </dd>
    1031. 

  1031. </dl>
    1032. 

  1032. </dd></dl>
    1033. 

  1033. 

  1034. <dl class="py method">
    1035. 

  1035. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_full_backward_hook">
    1036. 

  1036. <span class="sig-name descname"><span class="pre">register_full_backward_hook</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">hook</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Callable" title="(in Python v3.10)"><span class="pre">Callable</span></a><span class="p"><span class="pre">[</span></span><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.10)"><span class="pre">Tuple</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Tensor</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.10)"><span class="pre">Tuple</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Tensor</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Tensor</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">RemovableHandle</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_full_backward_hook" title="Permalink to this definition">¶</a></dt>
    1037. 

  1037. <dd><p>Registers a backward hook on the module.</p>
    1038. 

  1038. <p>The hook will be called every time the gradients with respect to module
    1039. 

  1039. inputs are computed. The hook should have the following signature:</p>
    1040. 

  1040. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="n">hook</span><span class="p">(</span><span class="n">module</span><span class="p">,</span> <span class="n">grad_input</span><span class="p">,</span> <span class="n">grad_output</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">tuple</span><span class="p">(</span><span class="n">Tensor</span><span class="p">)</span> <span class="ow">or</span> <span class="kc">None</span>
    1041. 

  1041. </pre></div>
    1042. 

  1042. </div>
    1043. 

  1043. <p>The <code class="xref py py-attr docutils literal notranslate"><span class="pre">grad_input</span></code> and <code class="xref py py-attr docutils literal notranslate"><span class="pre">grad_output</span></code> are tuples that contain the gradients
    1044. 

  1044. with respect to the inputs and outputs respectively. The hook should
    1045. 

  1045. not modify its arguments, but it can optionally return a new gradient with
    1046. 

  1046. respect to the input that will be used in place of <code class="xref py py-attr docutils literal notranslate"><span class="pre">grad_input</span></code> in
    1047. 

  1047. subsequent computations. <code class="xref py py-attr docutils literal notranslate"><span class="pre">grad_input</span></code> will only correspond to the inputs given
    1048. 

  1048. as positional arguments and all kwarg arguments are ignored. Entries
    1049. 

  1049. in <code class="xref py py-attr docutils literal notranslate"><span class="pre">grad_input</span></code> and <code class="xref py py-attr docutils literal notranslate"><span class="pre">grad_output</span></code> will be <code class="docutils literal notranslate"><span class="pre">None</span></code> for all non-Tensor
    1050. 

  1050. arguments.</p>
    1051. 

  1051. <p>For technical reasons, when this hook is applied to a Module, its forward function will
    1052. 

  1052. receive a view of each Tensor passed to the Module. Similarly the caller will receive a view
    1053. 

  1053. of each Tensor returned by the Module’s forward function.</p>
    1054. 

  1054. <div class="admonition warning">
    1055. 

  1055. <p class="admonition-title">Warning</p>
    1056. 

  1056. <p>Modifying inputs or outputs inplace is not allowed when using backward hooks and
    1057. 

  1057. will raise an error.</p>
    1058. 

  1058. </div>
    1059. 

  1059. <dl class="field-list simple">
    1060. 

  1060. <dt class="field-odd">Returns</dt>
    1061. 

  1061. <dd class="field-odd"><p>a handle that can be used to remove the added hook by calling
    1062. 

  1062. <code class="docutils literal notranslate"><span class="pre">handle.remove()</span></code></p>
    1063. 

  1063. </dd>
    1064. 

  1064. <dt class="field-even">Return type</dt>
    1065. 

  1065. <dd class="field-even"><p><code class="xref py py-class docutils literal notranslate"><span class="pre">torch.utils.hooks.RemovableHandle</span></code></p>
    1066. 

  1066. </dd>
    1067. 

  1067. </dl>
    1068. 

  1068. </dd></dl>
    1069. 

  1069. 

  1070. <dl class="py method">
    1071. 

  1071. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_load_state_dict_post_hook">
    1072. 

  1072. <span class="sig-name descname"><span class="pre">register_load_state_dict_post_hook</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">hook</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_load_state_dict_post_hook" title="Permalink to this definition">¶</a></dt>
    1073. 

  1073. <dd><p>Registers a post hook to be run after module’s <code class="docutils literal notranslate"><span class="pre">load_state_dict</span></code>
    1074. 

  1074. is called.</p>
    1075. 

  1075. <dl class="simple">
    1076. 

  1076. <dt>It should have the following signature::</dt><dd><p>hook(module, incompatible_keys) -&gt; None</p>
    1077. 

  1077. </dd>
    1078. 

  1078. </dl>
    1079. 

  1079. <p>The <code class="docutils literal notranslate"><span class="pre">module</span></code> argument is the current module that this hook is registered
    1080. 

  1080. on, and the <code class="docutils literal notranslate"><span class="pre">incompatible_keys</span></code> argument is a <code class="docutils literal notranslate"><span class="pre">NamedTuple</span></code> consisting
    1081. 

  1081. of attributes <code class="docutils literal notranslate"><span class="pre">missing_keys</span></code> and <code class="docutils literal notranslate"><span class="pre">unexpected_keys</span></code>. <code class="docutils literal notranslate"><span class="pre">missing_keys</span></code>
    1082. 

  1082. is a <code class="docutils literal notranslate"><span class="pre">list</span></code> of <code class="docutils literal notranslate"><span class="pre">str</span></code> containing the missing keys and
    1083. 

  1083. <code class="docutils literal notranslate"><span class="pre">unexpected_keys</span></code> is a <code class="docutils literal notranslate"><span class="pre">list</span></code> of <code class="docutils literal notranslate"><span class="pre">str</span></code> containing the unexpected keys.</p>
    1084. 

  1084. <p>The given incompatible_keys can be modified inplace if needed.</p>
    1085. 

  1085. <p>Note that the checks performed when calling <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict"><code class="xref py py-func docutils literal notranslate"><span class="pre">load_state_dict()</span></code></a> with
    1086. 

  1086. <code class="docutils literal notranslate"><span class="pre">strict=True</span></code> are affected by modifications the hook makes to
    1087. 

  1087. <code class="docutils literal notranslate"><span class="pre">missing_keys</span></code> or <code class="docutils literal notranslate"><span class="pre">unexpected_keys</span></code>, as expected. Additions to either
    1088. 

  1088. set of keys will result in an error being thrown when <code class="docutils literal notranslate"><span class="pre">strict=True</span></code>, and
    1089. 

  1089. clearning out both missing and unexpected keys will avoid an error.</p>
    1090. 

  1090. <dl class="field-list simple">
    1091. 

  1091. <dt class="field-odd">Returns</dt>
    1092. 

  1092. <dd class="field-odd"><p>a handle that can be used to remove the added hook by calling
    1093. 

  1093. <code class="docutils literal notranslate"><span class="pre">handle.remove()</span></code></p>
    1094. 

  1094. </dd>
    1095. 

  1095. <dt class="field-even">Return type</dt>
    1096. 

  1096. <dd class="field-even"><p><code class="xref py py-class docutils literal notranslate"><span class="pre">torch.utils.hooks.RemovableHandle</span></code></p>
    1097. 

  1097. </dd>
    1098. 

  1098. </dl>
    1099. 

  1099. </dd></dl>
    1100. 

  1100. 

  1101. <dl class="py method">
    1102. 

  1102. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_module">
    1103. 

  1103. <span class="sig-name descname"><span class="pre">register_module</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">module</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Module</span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_module" title="Permalink to this definition">¶</a></dt>
    1104. 

  1104. <dd><p>Alias for <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.add_module"><code class="xref py py-func docutils literal notranslate"><span class="pre">add_module()</span></code></a>.</p>
    1105. 

  1105. </dd></dl>
    1106. 

  1106. 

  1107. <dl class="py method">
    1108. 

  1108. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_parameter">
    1109. 

  1109. <span class="sig-name descname"><span class="pre">register_parameter</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">name</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">param</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><span class="pre">Parameter</span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.register_parameter" title="Permalink to this definition">¶</a></dt>
    1110. 

  1110. <dd><p>Adds a parameter to the module.</p>
    1111. 

  1111. <p>The parameter can be accessed as an attribute using given name.</p>
    1112. 

  1112. <dl class="field-list simple">
    1113. 

  1113. <dt class="field-odd">Parameters</dt>
    1114. 

  1114. <dd class="field-odd"><ul class="simple">
    1115. 

  1115. <li><p><strong>name</strong> (<em>string</em>) – name of the parameter. The parameter can be accessed
    1116. 

  1116. from this module using the given name</p></li>
    1117. 

  1117. <li><p><strong>param</strong> (<em>Parameter</em><em> or </em><em>None</em>) – parameter to be added to the module. If
    1118. 

  1118. <code class="docutils literal notranslate"><span class="pre">None</span></code>, then operations that run on parameters, such as <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.cuda"><code class="xref py py-attr docutils literal notranslate"><span class="pre">cuda</span></code></a>,
    1119. 

  1119. are ignored. If <code class="docutils literal notranslate"><span class="pre">None</span></code>, the parameter is <strong>not</strong> included in the
    1120. 

  1120. module’s <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict"><code class="xref py py-attr docutils literal notranslate"><span class="pre">state_dict</span></code></a>.</p></li>
    1121. 

  1121. </ul>
    1122. 

  1122. </dd>
    1123. 

  1123. </dl>
    1124. 

  1124. </dd></dl>
    1125. 

  1125. 

  1126. <dl class="py method">
    1127. 

  1127. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.requires_grad_">
    1128. 

  1128. <span class="sig-name descname"><span class="pre">requires_grad_</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">requires_grad</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.requires_grad_" title="Permalink to this definition">¶</a></dt>
    1129. 

  1129. <dd><p>Change if autograd should record operations on parameters in this
    1130. 

  1130. module.</p>
    1131. 

  1131. <p>This method sets the parameters’ <code class="xref py py-attr docutils literal notranslate"><span class="pre">requires_grad</span></code> attributes
    1132. 

  1132. in-place.</p>
    1133. 

  1133. <p>This method is helpful for freezing part of the module for finetuning
    1134. 

  1134. or training parts of a model individually (e.g., GAN training).</p>
    1135. 

  1135. <p>See <span class="xref std std-ref">locally-disable-grad-doc</span> for a comparison between
    1136. 

  1136. <cite>.requires_grad_()</cite> and several similar mechanisms that may be confused with it.</p>
    1137. 

  1137. <dl class="field-list simple">
    1138. 

  1138. <dt class="field-odd">Parameters</dt>
    1139. 

  1139. <dd class="field-odd"><p><strong>requires_grad</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – whether autograd should record operations on
    1140. 

  1140. parameters in this module. Default: <code class="docutils literal notranslate"><span class="pre">True</span></code>.</p>
    1141. 

  1141. </dd>
    1142. 

  1142. <dt class="field-even">Returns</dt>
    1143. 

  1143. <dd class="field-even"><p>self</p>
    1144. 

  1144. </dd>
    1145. 

  1145. <dt class="field-odd">Return type</dt>
    1146. 

  1146. <dd class="field-odd"><p>Module</p>
    1147. 

  1147. </dd>
    1148. 

  1148. </dl>
    1149. 

  1149. </dd></dl>
    1150. 

  1150. 

  1151. <dl class="py method">
    1152. 

  1152. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.sample_input">
    1153. 

  1153. <span class="sig-name descname"><span class="pre">sample_input</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Any" title="(in Python v3.10)"><span class="pre">Any</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.sample_input" title="Permalink to this definition">¶</a></dt>
    1154. 

  1154. <dd><dl class="field-list simple">
    1155. 

  1155. <dt class="field-odd">Returns</dt>
    1156. 

  1156. <dd class="field-odd"><p></p>
    1157. 

  1157. </dd>
    1158. 

  1158. <dt class="field-even">Return type</dt>
    1159. 

  1159. <dd class="field-even"><p></p>
    1160. 

  1160. </dd>
    1161. 

  1161. </dl>
    1162. 

  1162. </dd></dl>
    1163. 

  1163. 

  1164. <dl class="py method">
    1165. 

  1165. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.set_extra_state">
    1166. 

  1166. <span class="sig-name descname"><span class="pre">set_extra_state</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">state</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Any" title="(in Python v3.10)"><span class="pre">Any</span></a></span></em><span class="sig-paren">)</span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.set_extra_state" title="Permalink to this definition">¶</a></dt>
    1167. 

  1167. <dd><p>This function is called from <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.load_state_dict"><code class="xref py py-func docutils literal notranslate"><span class="pre">load_state_dict()</span></code></a> to handle any extra state
    1168. 

  1168. found within the <cite>state_dict</cite>. Implement this function and a corresponding
    1169. 

  1169. <a class="reference internal" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_extra_state" title="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.get_extra_state"><code class="xref py py-func docutils literal notranslate"><span class="pre">get_extra_state()</span></code></a> for your module if you need to store extra state within its
    1170. 

  1170. <cite>state_dict</cite>.</p>
    1171. 

  1171. <dl class="field-list simple">
    1172. 

  1172. <dt class="field-odd">Parameters</dt>
    1173. 

  1173. <dd class="field-odd"><p><strong>state</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#dict" title="(in Python v3.10)"><em>dict</em></a>) – Extra state from the <cite>state_dict</cite></p>
    1174. 

  1174. </dd>
    1175. 

  1175. </dl>
    1176. 

  1176. </dd></dl>
    1177. 

  1177. 

  1178. <dl class="py method">
    1179. 

  1179. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.share_memory">
    1180. 

  1180. <span class="sig-name descname"><span class="pre">share_memory</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.share_memory" title="Permalink to this definition">¶</a></dt>
    1181. 

  1181. <dd><p>See <code class="xref py py-meth docutils literal notranslate"><span class="pre">torch.Tensor.share_memory_()</span></code></p>
    1182. 

  1182. </dd></dl>
    1183. 

  1183. 

  1184. <dl class="py method">
    1185. 

  1185. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict">
    1186. 

  1186. <span class="sig-name descname"><span class="pre">state_dict</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">destination</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">prefix</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">''</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">keep_vars</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.state_dict" title="Permalink to this definition">¶</a></dt>
    1187. 

  1187. <dd><p>Returns a dictionary containing a whole state of the module.</p>
    1188. 

  1188. <p>Both parameters and persistent buffers (e.g. running averages) are
    1189. 

  1189. included. Keys are corresponding parameter and buffer names.
    1190. 

  1190. Parameters and buffers set to <code class="docutils literal notranslate"><span class="pre">None</span></code> are not included.</p>
    1191. 

  1191. <div class="admonition warning">
    1192. 

  1192. <p class="admonition-title">Warning</p>
    1193. 

  1193. <p>Currently <code class="docutils literal notranslate"><span class="pre">state_dict()</span></code> also accepts positional arguments for
    1194. 

  1194. <code class="docutils literal notranslate"><span class="pre">destination</span></code>, <code class="docutils literal notranslate"><span class="pre">prefix</span></code> and <code class="docutils literal notranslate"><span class="pre">keep_vars</span></code> in order. However,
    1195. 

  1195. this is being deprecated and keyword arguments will be enforced in
    1196. 

  1196. future releases.</p>
    1197. 

  1197. </div>
    1198. 

  1198. <div class="admonition warning">
    1199. 

  1199. <p class="admonition-title">Warning</p>
    1200. 

  1200. <p>Please avoid the use of argument <code class="docutils literal notranslate"><span class="pre">destination</span></code> as it is not
    1201. 

  1201. designed for end-users.</p>
    1202. 

  1202. </div>
    1203. 

  1203. <dl class="field-list simple">
    1204. 

  1204. <dt class="field-odd">Parameters</dt>
    1205. 

  1205. <dd class="field-odd"><ul class="simple">
    1206. 

  1206. <li><p><strong>destination</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#dict" title="(in Python v3.10)"><em>dict</em></a><em>, </em><em>optional</em>) – If provided, the state of module will
    1207. 

  1207. be updated into the dict and the same object is returned.
    1208. 

  1208. Otherwise, an <code class="docutils literal notranslate"><span class="pre">OrderedDict</span></code> will be created and returned.
    1209. 

  1209. Default: <code class="docutils literal notranslate"><span class="pre">None</span></code>.</p></li>
    1210. 

  1210. <li><p><strong>prefix</strong> (<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><em>str</em></a><em>, </em><em>optional</em>) – a prefix added to parameter and buffer
    1211. 

  1211. names to compose the keys in state_dict. Default: <code class="docutils literal notranslate"><span class="pre">''</span></code>.</p></li>
    1212. 

  1212. <li><p><strong>keep_vars</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a><em>, </em><em>optional</em>) – by default the <code class="xref py py-class docutils literal notranslate"><span class="pre">Tensor</span></code> s
    1213. 

  1213. returned in the state dict are detached from autograd. If it’s
    1214. 

  1214. set to <code class="docutils literal notranslate"><span class="pre">True</span></code>, detaching will not be performed.
    1215. 

  1215. Default: <code class="docutils literal notranslate"><span class="pre">False</span></code>.</p></li>
    1216. 

  1216. </ul>
    1217. 

  1217. </dd>
    1218. 

  1218. <dt class="field-even">Returns</dt>
    1219. 

  1219. <dd class="field-even"><p>a dictionary containing a whole state of the module</p>
    1220. 

  1220. </dd>
    1221. 

  1221. <dt class="field-odd">Return type</dt>
    1222. 

  1222. <dd class="field-odd"><p><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#dict" title="(in Python v3.10)">dict</a></p>
    1223. 

  1223. </dd>
    1224. 

  1224. </dl>
    1225. 

  1225. <p>Example:</p>
    1226. 

  1226. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="n">module</span><span class="o">.</span><span class="n">state_dict</span><span class="p">()</span><span class="o">.</span><span class="n">keys</span><span class="p">()</span>
    1227. 

  1227. <span class="go">[&#39;bias&#39;, &#39;weight&#39;]</span>
    1228. 

  1228. </pre></div>
    1229. 

  1229. </div>
    1230. 

  1230. </dd></dl>
    1231. 

  1231. 

  1232. <dl class="py method">
    1233. 

  1233. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to">
    1234. 

  1234. <span class="sig-name descname"><span class="pre">to</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</span></span></em>, <em class="sig-param"><span class="o"><span class="pre">**</span></span><span class="n"><span class="pre">kwargs</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to" title="Permalink to this definition">¶</a></dt>
    1235. 

  1235. <dd><p>Moves and/or casts the parameters and buffers.</p>
    1236. 

  1236. <p>This can be called as</p>
    1237. 

  1237. <dl class="py function">
    1238. 

  1238. <dt class="sig sig-object py">
    1239. 

  1239. <span class="sig-name descname"><span class="pre">to</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">non_blocking</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span></dt>
    1240. 

  1240. <dd></dd></dl>
    1241. 

  1241. 

  1242. <dl class="py function">
    1243. 

  1243. <dt class="sig sig-object py">
    1244. 

  1244. <span class="sig-name descname"><span class="pre">to</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">dtype</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">non_blocking</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span></dt>
    1245. 

  1245. <dd></dd></dl>
    1246. 

  1246. 

  1247. <dl class="py function">
    1248. 

  1248. <dt class="sig sig-object py">
    1249. 

  1249. <span class="sig-name descname"><span class="pre">to</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">tensor</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">non_blocking</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span></dt>
    1250. 

  1250. <dd></dd></dl>
    1251. 

  1251. 

  1252. <dl class="py function">
    1253. 

  1253. <dt class="sig sig-object py">
    1254. 

  1254. <span class="sig-name descname"><span class="pre">to</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">memory_format</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">torch.channels_last</span></span></em><span class="sig-paren">)</span></dt>
    1255. 

  1255. <dd></dd></dl>
    1256. 

  1256. 

  1257. <p>Its signature is similar to <code class="xref py py-meth docutils literal notranslate"><span class="pre">torch.Tensor.to()</span></code>, but only accepts
    1258. 

  1258. floating point or complex <code class="xref py py-attr docutils literal notranslate"><span class="pre">dtype</span></code>s. In addition, this method will
    1259. 

  1259. only cast the floating point or complex parameters and buffers to <code class="xref py py-attr docutils literal notranslate"><span class="pre">dtype</span></code>
    1260. 

  1260. (if given). The integral parameters and buffers will be moved
    1261. 

  1261. <code class="xref py py-attr docutils literal notranslate"><span class="pre">device</span></code>, if that is given, but with dtypes unchanged. When
    1262. 

  1262. <code class="xref py py-attr docutils literal notranslate"><span class="pre">non_blocking</span></code> is set, it tries to convert/move asynchronously
    1263. 

  1263. with respect to the host if possible, e.g., moving CPU Tensors with
    1264. 

  1264. pinned memory to CUDA devices.</p>
    1265. 

  1265. <p>See below for examples.</p>
    1266. 

  1266. <div class="admonition note">
    1267. 

  1267. <p class="admonition-title">Note</p>
    1268. 

  1268. <p>This method modifies the module in-place.</p>
    1269. 

  1269. </div>
    1270. 

  1270. <dl class="field-list simple">
    1271. 

  1271. <dt class="field-odd">Parameters</dt>
    1272. 

  1272. <dd class="field-odd"><ul class="simple">
    1273. 

  1273. <li><p><strong>device</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">torch.device</span></code>) – the desired device of the parameters
    1274. 

  1274. and buffers in this module</p></li>
    1275. 

  1275. <li><p><strong>dtype</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">torch.dtype</span></code>) – the desired floating point or complex dtype of
    1276. 

  1276. the parameters and buffers in this module</p></li>
    1277. 

  1277. <li><p><strong>tensor</strong> (<em>torch.Tensor</em>) – Tensor whose dtype and device are the desired
    1278. 

  1278. dtype and device for all parameters and buffers in this module</p></li>
    1279. 

  1279. <li><p><strong>memory_format</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">torch.memory_format</span></code>) – the desired memory
    1280. 

  1280. format for 4D parameters and buffers in this module (keyword
    1281. 

  1281. only argument)</p></li>
    1282. 

  1282. </ul>
    1283. 

  1283. </dd>
    1284. 

  1284. <dt class="field-even">Returns</dt>
    1285. 

  1285. <dd class="field-even"><p>self</p>
    1286. 

  1286. </dd>
    1287. 

  1287. <dt class="field-odd">Return type</dt>
    1288. 

  1288. <dd class="field-odd"><p>Module</p>
    1289. 

  1289. </dd>
    1290. 

  1290. </dl>
    1291. 

  1291. <p>Examples:</p>
    1292. 

  1292. <div class="highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span>
    1293. 

  1293. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">weight</span>
    1294. 

  1294. <span class="go">Parameter containing:</span>
    1295. 

  1295. <span class="go">tensor([[ 0.1913, -0.3420],</span>
    1296. 

  1296. <span class="go">        [-0.5113, -0.2325]])</span>
    1297. 

  1297. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">double</span><span class="p">)</span>
    1298. 

  1298. <span class="go">Linear(in_features=2, out_features=2, bias=True)</span>
    1299. 

  1299. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">weight</span>
    1300. 

  1300. <span class="go">Parameter containing:</span>
    1301. 

  1301. <span class="go">tensor([[ 0.1913, -0.3420],</span>
    1302. 

  1302. <span class="go">        [-0.5113, -0.2325]], dtype=torch.float64)</span>
    1303. 

  1303. <span class="gp">&gt;&gt;&gt; </span><span class="n">gpu1</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">device</span><span class="p">(</span><span class="s2">&quot;cuda:1&quot;</span><span class="p">)</span>
    1304. 

  1304. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="n">gpu1</span><span class="p">,</span> <span class="n">dtype</span><span class="o">=</span><span class="n">torch</span><span class="o">.</span><span class="n">half</span><span class="p">,</span> <span class="n">non_blocking</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
    1305. 

  1305. <span class="go">Linear(in_features=2, out_features=2, bias=True)</span>
    1306. 

  1306. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">weight</span>
    1307. 

  1307. <span class="go">Parameter containing:</span>
    1308. 

  1308. <span class="go">tensor([[ 0.1914, -0.3420],</span>
    1309. 

  1309. <span class="go">        [-0.5112, -0.2324]], dtype=torch.float16, device=&#39;cuda:1&#39;)</span>
    1310. 

  1310. <span class="gp">&gt;&gt;&gt; </span><span class="n">cpu</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">device</span><span class="p">(</span><span class="s2">&quot;cpu&quot;</span><span class="p">)</span>
    1311. 

  1311. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="n">cpu</span><span class="p">)</span>
    1312. 

  1312. <span class="go">Linear(in_features=2, out_features=2, bias=True)</span>
    1313. 

  1313. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">weight</span>
    1314. 

  1314. <span class="go">Parameter containing:</span>
    1315. 

  1315. <span class="go">tensor([[ 0.1914, -0.3420],</span>
    1316. 

  1316. <span class="go">        [-0.5112, -0.2324]], dtype=torch.float16)</span>
    1317. 

  1317. 

  1318. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span> <span class="o">=</span> <span class="n">nn</span><span class="o">.</span><span class="n">Linear</span><span class="p">(</span><span class="mi">2</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="n">bias</span><span class="o">=</span><span class="kc">None</span><span class="p">)</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">cdouble</span><span class="p">)</span>
    1319. 

  1319. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="o">.</span><span class="n">weight</span>
    1320. 

  1320. <span class="go">Parameter containing:</span>
    1321. 

  1321. <span class="go">tensor([[ 0.3741+0.j,  0.2382+0.j],</span>
    1322. 

  1322. <span class="go">        [ 0.5593+0.j, -0.4443+0.j]], dtype=torch.complex128)</span>
    1323. 

  1323. <span class="gp">&gt;&gt;&gt; </span><span class="n">linear</span><span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">ones</span><span class="p">(</span><span class="mi">3</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="n">dtype</span><span class="o">=</span><span class="n">torch</span><span class="o">.</span><span class="n">cdouble</span><span class="p">))</span>
    1324. 

  1324. <span class="go">tensor([[0.6122+0.j, 0.1150+0.j],</span>
    1325. 

  1325. <span class="go">        [0.6122+0.j, 0.1150+0.j],</span>
    1326. 

  1326. <span class="go">        [0.6122+0.j, 0.1150+0.j]], dtype=torch.complex128)</span>
    1327. 

  1327. </pre></div>
    1328. 

  1328. </div>
    1329. 

  1329. </dd></dl>
    1330. 

  1330. 

  1331. <dl class="py method">
    1332. 

  1332. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to_empty">
    1333. 

  1333. <span class="sig-name descname"><span class="pre">to_empty</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="o"><span class="pre">*</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">device</span><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.to_empty" title="Permalink to this definition">¶</a></dt>
    1334. 

  1334. <dd><p>Moves the parameters and buffers to the specified device without copying storage.</p>
    1335. 

  1335. <dl class="field-list simple">
    1336. 

  1336. <dt class="field-odd">Parameters</dt>
    1337. 

  1337. <dd class="field-odd"><p><strong>device</strong> (<code class="xref py py-class docutils literal notranslate"><span class="pre">torch.device</span></code>) – The desired device of the parameters
    1338. 

  1338. and buffers in this module.</p>
    1339. 

  1339. </dd>
    1340. 

  1340. <dt class="field-even">Returns</dt>
    1341. 

  1341. <dd class="field-even"><p>self</p>
    1342. 

  1342. </dd>
    1343. 

  1343. <dt class="field-odd">Return type</dt>
    1344. 

  1344. <dd class="field-odd"><p>Module</p>
    1345. 

  1345. </dd>
    1346. 

  1346. </dl>
    1347. 

  1347. </dd></dl>
    1348. 

  1348. 

  1349. <dl class="py method">
    1350. 

  1350. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.train">
    1351. 

  1351. <span class="sig-name descname"><span class="pre">train</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">mode</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">True</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.train" title="Permalink to this definition">¶</a></dt>
    1352. 

  1352. <dd><p>Sets the module in training mode.</p>
    1353. 

  1353. <p>This has any effect only on certain modules. See documentations of
    1354. 

  1354. particular modules for details of their behaviors in training/evaluation
    1355. 

  1355. mode, if they are affected, e.g. <code class="xref py py-class docutils literal notranslate"><span class="pre">Dropout</span></code>, <code class="xref py py-class docutils literal notranslate"><span class="pre">BatchNorm</span></code>,
    1356. 

  1356. etc.</p>
    1357. 

  1357. <dl class="field-list simple">
    1358. 

  1358. <dt class="field-odd">Parameters</dt>
    1359. 

  1359. <dd class="field-odd"><p><strong>mode</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – whether to set training mode (<code class="docutils literal notranslate"><span class="pre">True</span></code>) or evaluation
    1360. 

  1360. mode (<code class="docutils literal notranslate"><span class="pre">False</span></code>). Default: <code class="docutils literal notranslate"><span class="pre">True</span></code>.</p>
    1361. 

  1361. </dd>
    1362. 

  1362. <dt class="field-even">Returns</dt>
    1363. 

  1363. <dd class="field-even"><p>self</p>
    1364. 

  1364. </dd>
    1365. 

  1365. <dt class="field-odd">Return type</dt>
    1366. 

  1366. <dd class="field-odd"><p>Module</p>
    1367. 

  1367. </dd>
    1368. 

  1368. </dl>
    1369. 

  1369. </dd></dl>
    1370. 

  1370. 

  1371. <dl class="py method">
    1372. 

  1372. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.type">
    1373. 

  1373. <span class="sig-name descname"><span class="pre">type</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">dst_type</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><span class="pre">dtype</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.10)"><span class="pre">str</span></a><span class="p"><span class="pre">]</span></span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.type" title="Permalink to this definition">¶</a></dt>
    1374. 

  1374. <dd><p>Casts all parameters and buffers to <code class="xref py py-attr docutils literal notranslate"><span class="pre">dst_type</span></code>.</p>
    1375. 

  1375. <div class="admonition note">
    1376. 

  1376. <p class="admonition-title">Note</p>
    1377. 

  1377. <p>This method modifies the module in-place.</p>
    1378. 

  1378. </div>
    1379. 

  1379. <dl class="field-list simple">
    1380. 

  1380. <dt class="field-odd">Parameters</dt>
    1381. 

  1381. <dd class="field-odd"><p><strong>dst_type</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#type" title="(in Python v3.10)"><em>type</em></a><em> or </em><em>string</em>) – the desired type</p>
    1382. 

  1382. </dd>
    1383. 

  1383. <dt class="field-even">Returns</dt>
    1384. 

  1384. <dd class="field-even"><p>self</p>
    1385. 

  1385. </dd>
    1386. 

  1386. <dt class="field-odd">Return type</dt>
    1387. 

  1387. <dd class="field-odd"><p>Module</p>
    1388. 

  1388. </dd>
    1389. 

  1389. </dl>
    1390. 

  1390. </dd></dl>
    1391. 

  1391. 

  1392. <dl class="py method">
    1393. 

  1393. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.xpu">
    1394. 

  1394. <span class="sig-name descname"><span class="pre">xpu</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">device</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Optional" title="(in Python v3.10)"><span class="pre">Optional</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.10)"><span class="pre">Union</span></a><span class="p"><span class="pre">[</span></span><a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><span class="pre">int</span></a><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">device</span><span class="p"><span class="pre">]</span></span><span class="p"><span class="pre">]</span></span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">T</span></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.xpu" title="Permalink to this definition">¶</a></dt>
    1395. 

  1395. <dd><p>Moves all model parameters and buffers to the XPU.</p>
    1396. 

  1396. <p>This also makes associated parameters and buffers different objects. So
    1397. 

  1397. it should be called before constructing optimizer if the module will
    1398. 

  1398. live on XPU while being optimized.</p>
    1399. 

  1399. <div class="admonition note">
    1400. 

  1400. <p class="admonition-title">Note</p>
    1401. 

  1401. <p>This method modifies the module in-place.</p>
    1402. 

  1402. </div>
    1403. 

  1403. <dl class="field-list simple">
    1404. 

  1404. <dt class="field-odd">Parameters</dt>
    1405. 

  1405. <dd class="field-odd"><p><strong>device</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.10)"><em>int</em></a><em>, </em><em>optional</em>) – if specified, all parameters will be
    1406. 

  1406. copied to that device</p>
    1407. 

  1407. </dd>
    1408. 

  1408. <dt class="field-even">Returns</dt>
    1409. 

  1409. <dd class="field-even"><p>self</p>
    1410. 

  1410. </dd>
    1411. 

  1411. <dt class="field-odd">Return type</dt>
    1412. 

  1412. <dd class="field-odd"><p>Module</p>
    1413. 

  1413. </dd>
    1414. 

  1414. </dl>
    1415. 

  1415. </dd></dl>
    1416. 

  1416. 

  1417. <dl class="py method">
    1418. 

  1418. <dt class="sig sig-object py" id="draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.zero_grad">
    1419. 

  1419. <span class="sig-name descname"><span class="pre">zero_grad</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">set_to_none</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><span class="pre">bool</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">False</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.10)"><span class="pre">None</span></a></span></span><a class="headerlink" href="#draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.zero_grad" title="Permalink to this definition">¶</a></dt>
    1420. 

  1420. <dd><p>Sets gradients of all model parameters to zero. See similar function
    1421. 

  1421. under <code class="xref py py-class docutils literal notranslate"><span class="pre">torch.optim.Optimizer</span></code> for more context.</p>
    1422. 

  1422. <dl class="field-list simple">
    1423. 

  1423. <dt class="field-odd">Parameters</dt>
    1424. 

  1424. <dd class="field-odd"><p><strong>set_to_none</strong> (<a class="reference external" href="https://docs.python.org/3/library/functions.html#bool" title="(in Python v3.10)"><em>bool</em></a>) – instead of setting to zero, set the grads to None.
    1425. 

  1425. See <code class="xref py py-meth docutils literal notranslate"><span class="pre">torch.optim.Optimizer.zero_grad()</span></code> for details.</p>
    1426. 

  1426. </dd>
    1427. 

  1427. </dl>
    1428. 

  1428. </dd></dl>
    1429. 

  1429. 

  1430. </dd></dl>
    1431. 

  1431. 

  1432. </section>
    1433. 

  1433. 

  1434. 

  1435.           </div>
    1436. 

  1436.           
    1437. 

  1437.         </div>
    1438. 

  1438.       </div>
    1439. 

  1439.       <div class="sphinxsidebar" role="navigation" aria-label="main navigation">
    1440. 

  1440.         <div class="sphinxsidebarwrapper">
    1441. 

  1441.             <p class="logo"><a href="../index.html">
    1442. 

  1442.               <img class="logo" src="../_static/valknut.svg" alt="Logo"/>
    1443. 

  1443.             </a></p>
    1444. 

  1444. <h1 class="logo"><a href="../index.html">Draugr</a></h1>
    1445. 

  1445. 

  1446. 

  1447. 

  1448. 

  1449. 

  1450. 

  1451. 

  1452. 

  1453. <h3>Navigation</h3>
    1454. 

  1454. <ul class="current">
    1455. 

  1455. <li class="toctree-l1 current"><a class="reference internal" href="draugr.html">draugr</a><ul class="current">
    1456. 

  1456. <li class="toctree-l2"><a class="reference internal" href="draugr.dist_is_editable.html">draugr.dist_is_editable</a></li>
    1457. 

  1457. <li class="toctree-l2"><a class="reference internal" href="draugr.get_version.html">draugr.get_version</a></li>
    1458. 

  1458. <li class="toctree-l2"><a class="reference internal" href="draugr.dlib_utilities.html">draugr.dlib_utilities</a></li>
    1459. 

  1459. <li class="toctree-l2"><a class="reference internal" href="draugr.drawers.html">draugr.drawers</a></li>
    1460. 

  1460. <li class="toctree-l2"><a class="reference internal" href="draugr.entry_points.html">draugr.entry_points</a></li>
    1461. 

  1461. <li class="toctree-l2"><a class="reference internal" href="draugr.extensions.html">draugr.extensions</a></li>
    1462. 

  1462. <li class="toctree-l2"><a class="reference internal" href="draugr.ffmpeg_utilities.html">draugr.ffmpeg_utilities</a></li>
    1463. 

  1463. <li class="toctree-l2"><a class="reference internal" href="draugr.jax_utilities.html">draugr.jax_utilities</a></li>
    1464. 

  1464. <li class="toctree-l2"><a class="reference internal" href="draugr.metrics.html">draugr.metrics</a></li>
    1465. 

  1465. <li class="toctree-l2"><a class="reference internal" href="draugr.multiprocessing_utilities.html">draugr.multiprocessing_utilities</a></li>
    1466. 

  1466. <li class="toctree-l2"><a class="reference internal" href="draugr.numpy_utilities.html">draugr.numpy_utilities</a></li>
    1467. 

  1467. <li class="toctree-l2"><a class="reference internal" href="draugr.opencv_utilities.html">draugr.opencv_utilities</a></li>
    1468. 

  1468. <li class="toctree-l2"><a class="reference internal" href="draugr.os_utilities.html">draugr.os_utilities</a></li>
    1469. 

  1469. <li class="toctree-l2"><a class="reference internal" href="draugr.pandas_utilities.html">draugr.pandas_utilities</a></li>
    1470. 

  1470. <li class="toctree-l2"><a class="reference internal" href="draugr.pygame_utilities.html">draugr.pygame_utilities</a></li>
    1471. 

  1471. <li class="toctree-l2"><a class="reference internal" href="draugr.python_utilities.html">draugr.python_utilities</a></li>
    1472. 

  1472. <li class="toctree-l2"><a class="reference internal" href="draugr.random_utilities.html">draugr.random_utilities</a></li>
    1473. 

  1473. <li class="toctree-l2"><a class="reference internal" href="draugr.scipy_utilities.html">draugr.scipy_utilities</a></li>
    1474. 

  1474. <li class="toctree-l2"><a class="reference internal" href="draugr.stopping.html">draugr.stopping</a></li>
    1475. 

  1475. <li class="toctree-l2"><a class="reference internal" href="draugr.tensorboard_utilities.html">draugr.tensorboard_utilities</a></li>
    1476. 

  1476. <li class="toctree-l2"><a class="reference internal" href="draugr.threading_utilities.html">draugr.threading_utilities</a></li>
    1477. 

  1477. <li class="toctree-l2 current"><a class="reference internal" href="draugr.torch_utilities.html">draugr.torch_utilities</a><ul class="current">
    1478. 

  1478. <li class="toctree-l3 current"><a class="reference internal" href="draugr.torch_utilities.architectures.html">draugr.torch_utilities.architectures</a><ul class="current">
    1479. 

  1479. <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.architectures.architecture.html">draugr.torch_utilities.architectures.architecture</a></li>
    1480. 

  1480. <li class="toctree-l4 current"><a class="reference internal" href="draugr.torch_utilities.architectures.distributional.html">draugr.torch_utilities.architectures.distributional</a><ul class="current">
    1481. 

  1481. <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.architectures.distributional.categorical.html">draugr.torch_utilities.architectures.distributional.categorical</a></li>
    1482. 

  1482. <li class="toctree-l5 current"><a class="reference internal" href="draugr.torch_utilities.architectures.distributional.normal.html">draugr.torch_utilities.architectures.distributional.normal</a><ul class="current">
    1483. 

  1483. <li class="toctree-l6"><a class="reference internal" href="draugr.torch_utilities.architectures.distributional.normal.ShallowStdNormalMLP.html">draugr.torch_utilities.architectures.distributional.normal.ShallowStdNormalMLP</a></li>
    1484. 

  1484. <li class="toctree-l6"><a class="reference internal" href="draugr.torch_utilities.architectures.distributional.normal.MultiDimensionalNormalMLP.html">draugr.torch_utilities.architectures.distributional.normal.MultiDimensionalNormalMLP</a></li>
    1485. 

  1485. <li class="toctree-l6 current"><a class="current reference internal" href="#">draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP</a></li>
    1486. 

  1486. <li class="toctree-l6"><a class="reference internal" href="draugr.torch_utilities.architectures.distributional.normal.MultipleNormalMLP.html">draugr.torch_utilities.architectures.distributional.normal.MultipleNormalMLP</a></li>
    1487. 

  1487. </ul>
    1488. 

  1488. </li>
    1489. 

  1489. </ul>
    1490. 

  1490. </li>
    1491. 

  1491. <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.architectures.experimental.html">draugr.torch_utilities.architectures.experimental</a></li>
    1492. 

  1492. <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.architectures.mlp.html">draugr.torch_utilities.architectures.mlp</a></li>
    1493. 

  1493. <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.architectures.mlp_variants.html">draugr.torch_utilities.architectures.mlp_variants</a></li>
    1494. 

  1494. </ul>
    1495. 

  1495. </li>
    1496. 

  1496. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.datasets.html">draugr.torch_utilities.datasets</a></li>
    1497. 

  1497. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.distributions.html">draugr.torch_utilities.distributions</a></li>
    1498. 

  1498. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.evaluation.html">draugr.torch_utilities.evaluation</a></li>
    1499. 

  1499. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.exporting.html">draugr.torch_utilities.exporting</a></li>
    1500. 

  1500. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.generators.html">draugr.torch_utilities.generators</a></li>
    1501. 

  1501. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.images.html">draugr.torch_utilities.images</a></li>
    1502. 

  1502. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.opencv.html">draugr.torch_utilities.opencv</a></li>
    1503. 

  1503. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.operations.html">draugr.torch_utilities.operations</a></li>
    1504. 

  1504. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.optimisation.html">draugr.torch_utilities.optimisation</a></li>
    1505. 

  1505. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.persistence.html">draugr.torch_utilities.persistence</a></li>
    1506. 

  1506. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.sessions.html">draugr.torch_utilities.sessions</a></li>
    1507. 

  1507. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.system.html">draugr.torch_utilities.system</a></li>
    1508. 

  1508. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.tensors.html">draugr.torch_utilities.tensors</a></li>
    1509. 

  1509. <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.writers.html">draugr.torch_utilities.writers</a></li>
    1510. 

  1510. </ul>
    1511. 

  1511. </li>
    1512. 

  1512. <li class="toctree-l2"><a class="reference internal" href="draugr.tqdm_utilities.html">draugr.tqdm_utilities</a></li>
    1513. 

  1513. <li class="toctree-l2"><a class="reference internal" href="draugr.visualisation.html">draugr.visualisation</a></li>
    1514. 

  1514. <li class="toctree-l2"><a class="reference internal" href="draugr.writers.html">draugr.writers</a></li>
    1515. 

  1515. </ul>
    1516. 

  1516. </li>
    1517. 

  1517. </ul>
    1518. 

  1518. <p class="caption" role="heading"><span class="caption-text">Notes</span></p>
    1519. 

  1519. <ul>
    1520. 

  1520. <li class="toctree-l1"><a class="reference internal" href="../getting_started.html">Getting Started</a></li>
    1521. 

  1521. </ul>
    1522. 

  1522. 

  1523. <div class="relations">
    1524. 

  1524. <h3>Related Topics</h3>
    1525. 

  1525. <ul>
    1526. 

  1526.   <li><a href="../index.html">Documentation overview</a><ul>
    1527. 

  1527.   <li><a href="draugr.html">draugr</a><ul>
    1528. 

  1528.   <li><a href="draugr.torch_utilities.html">draugr.torch_utilities</a><ul>
    1529. 

  1529.   <li><a href="draugr.torch_utilities.architectures.html">draugr.torch_utilities.architectures</a><ul>
    1530. 

  1530.   <li><a href="draugr.torch_utilities.architectures.distributional.html">draugr.torch_utilities.architectures.distributional</a><ul>
    1531. 

  1531.   <li><a href="draugr.torch_utilities.architectures.distributional.normal.html">draugr.torch_utilities.architectures.distributional.normal</a><ul>
    1532. 

  1532.       <li>Previous: <a href="draugr.torch_utilities.architectures.distributional.normal.MultiDimensionalNormalMLP.html" title="previous chapter">draugr.torch_utilities.architectures.distributional.normal.MultiDimensionalNormalMLP</a></li>
    1533. 

  1533.       <li>Next: <a href="draugr.torch_utilities.architectures.distributional.normal.MultipleNormalMLP.html" title="next chapter">draugr.torch_utilities.architectures.distributional.normal.MultipleNormalMLP</a></li>
    1534. 

  1534.   </ul></li>
    1535. 

  1535.   </ul></li>
    1536. 

  1536.   </ul></li>
    1537. 

  1537.   </ul></li>
    1538. 

  1538.   </ul></li>
    1539. 

  1539.   </ul></li>
    1540. 

  1540. </ul>
    1541. 

  1541. </div>
    1542. 

  1542. <div id="searchbox" style="display: none" role="search">
    1543. 

  1543.   <h3 id="searchlabel">Quick search</h3>
    1544. 

  1544.     <div class="searchformwrapper">
    1545. 

  1545.     <form class="search" action="../search.html" method="get">
    1546. 

  1546.       <input type="text" name="q" aria-labelledby="searchlabel" autocomplete="off" autocorrect="off" autocapitalize="off" spellcheck="false"/>
    1547. 

  1547.       <input type="submit" value="Go" />
    1548. 

  1548.     </form>
    1549. 

  1549.     </div>
    1550. 

  1550. </div>
    1551. 

  1551. <script>document.getElementById('searchbox').style.display = "block"</script>
    1552. 

  1552. 

  1553. 

  1554. 

  1555. 

  1556. 

  1557. 

  1558. 

  1559. 

  1560.         </div>
    1561. 

  1561.       </div>
    1562. 

  1562.       <div class="clearer"></div>
    1563. 

  1563.     </div>
    1564. 

  1564.     <div class="footer">
    1565. 

  1565.       &copy;.
    1566. 

  1566.       
    1567. 

  1567.       |
    1568. 

  1568.       Powered by <a href="http://sphinx-doc.org/">Sphinx 5.0.2</a>
    1569. 

  1569.       &amp; <a href="https://github.com/bitprophet/alabaster">Alabaster 0.7.12</a>
    1570. 

  1570.       
    1571. 

  1571.       |
    1572. 

  1572.       <a href="../_sources/generated/draugr.torch_utilities.architectures.distributional.normal.MultiVariateNormalMLP.rst.txt"
    1573. 

  1573.           rel="nofollow">Page source</a>
    1574. 

  1574.     </div>
    1575. 

  1575. 

  1576.     
    1577. 

  1577. 

  1578.     
    1579. 

  1579.   </body>
    1580. 

  1580. </html>
    1581.