div class="ref-description">
    <p>Applies Batch Normalization (Ioffe and Szegedy, 2015) to samples from a
data distribution. This can be used to stabilize training of normalizing
flows (Papamakarios et al., 2016; Dinh et al., 2017)</p>
    </div>

    <pre class="usage"><span class='fu'>tfb_batch_normalization</span><span class='op'>(</span>
  batchnorm_layer <span class='op'>=</span> <span class='cn'>NULL</span>,
  training <span class='op'>=</span> <span class='cn'>TRUE</span>,
  validate_args <span cla