lass="o">.</span><span class="n">ToTensor</span><span class="p">(),</span>
                 <span class="n">transforms</span><span class="o">.</span><span class="n">Normalize</span><span class="p">((</span><span class="mf">0.1307</span><span class="p">,),</span> <span class="p">(</span><span class="mf">0.3081</span><span class="p">,))])),</span>
                 <span class="n">batch_size</span><span class="o">=</span><span class="mi">128</span><span class="p">,</span>
                 <span class="n">shuff