import torch


def prob_2_vr(prob):
    """ convert probabilistic prediction maps to weighted self-information maps
    """
    c, h, w = prob.size()
    # print(prob.shape)
    return 1 - torch.max(prob, dim=0)[0]