

    <span class="k">with</span> <span class="n">torch</span><span class="o">.</span><span class="n">no_grad</span><span class="p">():</span>
        <span class="n">output</span> <span class="o">=</span> <span class="n">model</span><span class="p">(</span><span class="n">input_batch</span><span class="p">)</span>
        <span class="c1"># Tensor of shape 1000, with confidence scores over Imagenet's 1000 classes</span>
        <span class="n">sm_output</span> <span class="o">=</span> <span class="n">torch