ss="kn">import</span> <span class="n">Sequence</span>

<span class="kn">import</span> <span class="nn">torch</span>
<span class="kn">from</span> <span class="nn">torch</span> <span class="kn">import</span> <span class="n">nn</span>


<div class="viewcode-block" id="IndependentGaussian"><a class="viewcode-back" href="../../../api/layers.html#torchdrug.layers.distribution.IndependentGaussian">[docs]</a><span class="k">class</span> <span class="nc">IndependentGaussian</span><span class="p">(</span><span class=