yunjey

def contextual_loss(x, y, h=0.5):
    """Computes contextual loss between x and y.
    
    Args:
      x: features of shape (N, C, H, W).
      y: features of shape (N, C, H, W).
      
    Returns:
      cx_loss = contextual loss between x and y (Eq (1) in the paper)
    """