pan> <span class="k">const</span> <span class="k">override</span><span class="p">;</span>

    <span class="c1">// The mode the dataset is in</span>
    <span class="kt">bool</span> <span class="nf">is_train</span><span class="p">()</span> <span class="k">const</span> <span class="k">noexcept</span><span class="p">;</span>

    <span class="c1">// Returns all images stacked into a single tensor</span>
    <span class="k">const</span> <span class="n">torch</span><span class="o">::</span><span class="n">Tenso