u</code></pre>
</details>
</section>
<section>
</section>
<section>
</section>
<section>
<h2 class="section-title" id="header-functions">Functions</h2>
<dl>
<dt id="flow.utils.log_mean_exp_trick"><code class="name flex">
<span>def <span class="ident">log_mean_exp_trick</span></span>(<span>x, dim=-1, keepdim=False)</span>
</code></dt>
<dd>
<div class="desc"><p>Computes log(mean(exp(x), dim=dim, keepdim=keepdim)) safely.</p>
<p>Uses the logsumexp trick for the computation of this quantity.</p></div>
<details 