def __init__(self, reduction='mean'):
    super(GMMLoss, self).__init__()
    self.reduction = reduction
def __init__(self, reduction='mean'):
    super(GMMLoss, self).__init__()
    self.reduction = reduction