code: def binaryEntropy(x):
    
    entropy = - x*x.log2() - (1-x)*(1-x).log2()
    entropy[x*(1 - x) == 0] = 0
    return entropy, entropy.sum()
docs: Calculate entropy for a list of binary random variables
  
    :param x: (torch tensor) the probability of the variable to be 1.
    :return: entropy: (torch tensor) entropy, sum(entropy)