def METHOD_NAME(self):
    """
    Returns the variance of the distribution, of shape (*batch_shape,)
    """
    xi, beta = self.xi, self.beta
    var = torch.where(
        xi < 1 / 2.0,
        torch.div(beta**2, torch.mul((1 - xi) ** 2, (1 - 2 * xi))),
        np.nan * torch.ones_like(xi),
    )
    return var