how to get the confidence score from the output result? about dsntnn HOT 2 OPEN

def heatmap_variance(heatmaps):
    # copied and modified:
    # https://github.com/anibali/dsntnn/blob/4f20f5a85b56d007adef51e5158f5a6dca92794f/dsntnn/__init__.py#L233-L262

    # mu = E[X]
    values = [normalized_linspace(d, dtype=heatmaps.dtype, device=heatmaps.device)  for d in heatmaps.size()[2:]]
    mu = linear_expectation(heatmaps, values)

    # var = E[(X - mu)^2]
    values = [(a - b.squeeze(0)) ** 2 for a, b in zip(values, mu.split(1, -1))]
    var = linear_expectation(heatmaps, values)

    heatmap_size = torch.tensor(list(heatmaps.size()[2:]), dtype=var.dtype, device=var.device)

    return var * (heatmap_size / 2) ** 2