ass="o">::</span><span class="n">string</span><span class="o">&amp;</span> <span class="n">root</span><span class="p">,</span> <span class="n">Mode</span> <span class="n">mode</span> <span class="o">=</span> <span class="n">Mode</span><span class="o">::</span><span class="n">kTrain</span><span class="p">);</span>

    <span class="c1">// Returns the pair at index in the dataset</span>
    <span class="n">torch</span><span class="o">::</span><span class="n">data</span><span class="o">::</span><span class="n"