="str">        Args:</span>&nbsp;</span><span class="r"></span></p>
    <p class="pln"><span class="n"><a id="t17" href="#t17">17</a></span><span class="t"><span class="str">            pred (torch.Tensor): The prediction tensor.</span>&nbsp;</span><span class="r"></span></p>
    <p class="pln"><span class="n"><a id="t18" href="#t18">18</a></span><span class="t"><span class="str">            gt (torch.Tensor): The ground-truth tensor.</span>&nbsp;</span><span class="r"></span></p>
    <p class="pln"><span c