(self, pred: torch.Tensor, gt, mask, weights=None)
return self._compute(pred, gt, mask, weights)
pred: one or two heatmaps of shape (N, 1, H, W),
            the losses of tow heatmaps are added together.
        gt: (N, 1, H, W)
        mask: (N, H, W)
pred: one or two heatmaps of shape (N, 1, H, W),
            the losses of tow heatmaps are added together.
        gt: (N, 1, H, W)
        mask: (N, H, W)
pred : one or two heatmaps of shape ( N 1 H W ) the losses of tow heatmaps are added together . gt : ( N 1 H W ) mask : ( N H W )
def forward(self, pred: torch.Tensor, gt, mask, weights=None):
        '''
        pred: one or two heatmaps of shape (N, 1, H, W),
            the losses of tow heatmaps are added together.
        gt: (N, 1, H, W)
        mask: (N, H, W)
        '''
        return self._compute(pred, gt, mask, weights)
def forward ( self , pred : torch . Tensor , gt , mask , weights = None ) : return self . _compute ( pred , gt , mask , weights )