"n">nn</span><span class="o">.</span><span class="n">Module</span><span class="p">):</span>
    <span class="sd">"""</span>
<span class="sd">    Independent Gaussian distribution.</span>

<span class="sd">    Parameters:</span>
<span class="sd">        mu (Tensor): mean of shape :math:`(N,)`</span>
<span class="sd">        sigma2 (Tensor): variance of shape :math:`(N,)`</span>
<span class="sd">        learnable (bool, optional): learnable parameters or not</span>
<span class="sd">    """</span>

    <span c