e dataset to the first n pairs</span>
    <span class="n">CIFAR10</span><span class="o">&amp;&amp;</span> <span class="n">use_subset</span><span class="p">(</span><span class="kt">int64_t</span> <span class="n">new_size</span><span class="p">);</span>


<span class="k">private</span><span class="o">:</span>
    <span class="n">Mode</span> <span class="n">mode_</span><span class="p">;</span>
    <span class="n">torch</span><span class="o">::</span><span class="n">Tensor</span> <span class="n">images_</span><