shape</span><span class="o">=</span><span class="p">[</span><span class="n">batch_size</span><span class="p">,</span> <span class="n">c</span><span class="p">,</span> <span class="n">h</span> <span class="o">//</span> <span class="n">f</span><span class="p">,</span> <span class="n">w</span> <span class="o">//</span> <span class="n">f</span><span class="p">],</span>
<span class="lineno">95</span>                                    <span class="n">uncond_scale</span><span class="o">=</span><span class="n">unc